28 namespace experimental
30 namespace dynamic_fusion
37 return ckw::DataType::Fp32;
39 return ckw::DataType::Fp16;
41 return ckw::DataType::Int32;
43 return ckw::DataType::Int16;
46 return ckw::DataType::Int8;
48 return ckw::DataType::Uint32;
50 return ckw::DataType::Uint16;
53 return ckw::DataType::Uint8;
55 return ckw::DataType::Unknown;
64 return ckw::TensorShape{
static_cast<int32_t
>(
shape[0]),
static_cast<int32_t
>(
shape[1]),
65 static_cast<int32_t
>(
shape[2]),
static_cast<int32_t
>(
shape[3]),
66 static_cast<int32_t
>(
shape[4])};
74 return ckw::TensorDataLayout::Nhwc;
76 return ckw::TensorDataLayout::Ndhwc;
78 return ckw::TensorDataLayout::Unknown;
88 ckw::TensorStorageType
to_ckw(
const TensorStorageType &storage)
92 case TensorStorageType::ClBufferUint8Ptr:
93 return ckw::TensorStorageType::BufferUint8Ptr;
94 case TensorStorageType::ClImage2dReadOnly:
95 return ckw::TensorStorageType::Texture2dReadOnly;
96 case TensorStorageType::ClImage2dWriteOnly:
97 return ckw::TensorStorageType::Texture2dWriteOnly;
98 case TensorStorageType::Unknown:
99 return ckw::TensorStorageType::Unknown;
105 TensorComponentType
from_ckw(
const ckw::TensorComponentType &component)
109 case ckw::TensorComponentType::OffsetFirstElement:
110 return TensorComponentType::OffsetFirstElement;
111 case ckw::TensorComponentType::Stride0:
112 return TensorComponentType::Stride0;
113 case ckw::TensorComponentType::Stride1:
114 return TensorComponentType::Stride1;
115 case ckw::TensorComponentType::Stride2:
116 return TensorComponentType::Stride2;
117 case ckw::TensorComponentType::Stride3:
118 return TensorComponentType::Stride3;
119 case ckw::TensorComponentType::Stride4:
120 return TensorComponentType::Stride4;
121 case ckw::TensorComponentType::Dim0:
122 return TensorComponentType::Dim0;
123 case ckw::TensorComponentType::Dim1:
124 return TensorComponentType::Dim1;
125 case ckw::TensorComponentType::Dim2:
126 return TensorComponentType::Dim2;
127 case ckw::TensorComponentType::Dim3:
128 return TensorComponentType::Dim3;
129 case ckw::TensorComponentType::Dim4:
130 return TensorComponentType::Dim4;
131 case ckw::TensorComponentType::Dim1xDim2:
132 return TensorComponentType::Dim1xDim2;
133 case ckw::TensorComponentType::Dim2xDim3:
134 return TensorComponentType::Dim2xDim3;
135 case ckw::TensorComponentType::Dim1xDim2xDim3:
136 return TensorComponentType::Dim1xDim2xDim3;
137 case ckw::TensorComponentType::Unknown:
138 return TensorComponentType::Unknown;
144 TensorStorageType
from_ckw(
const ckw::TensorStorageType &storage)
148 case ckw::TensorStorageType::BufferUint8Ptr:
149 return TensorStorageType::ClBufferUint8Ptr;
150 case ckw::TensorStorageType::Texture2dReadOnly:
151 return TensorStorageType::ClImage2dReadOnly;
152 case ckw::TensorStorageType::Texture2dWriteOnly:
153 return TensorStorageType::ClImage2dWriteOnly;
154 case ckw::TensorStorageType::Unknown:
155 return TensorStorageType::Unknown;