device_tensor.hpp 257 Bytes
Newer Older
Chao Liu's avatar
Chao Liu committed
1
#pragma once
Chao Liu's avatar
Chao Liu committed
2
#include "host_tensor.hpp"
Chao Liu's avatar
Chao Liu committed
3
4
#include "common_header.hpp"

Chao Liu's avatar
Chao Liu committed
5
6
template <typename TensorDesc>
void ostream_tensor_descriptor(TensorDesc, std::ostream& os = std::cout)
Chao Liu's avatar
Chao Liu committed
7
{
Chao Liu's avatar
Chao Liu committed
8
    ostream_HostTensorDescriptor(make_HostTensorDescriptor(TensorDesc{}), os);
Chao Liu's avatar
Chao Liu committed
9
}