Searched refs:HWDC (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/ |
| D | tensor_test.cc | 111 BHWDC(1, 6, 7, 4, 3), {data_type, storage_type, Layout::HWDC}, env)); in TensorTests()
113 BHWDC(1, 1, 4, 3, 12), {data_type, storage_type, Layout::HWDC}, env)); in TensorTests()
115 BHWDC(1, 6, 1, 7, 7), {data_type, storage_type, Layout::HWDC}, env)); in TensorTests()
200 {DataType::FLOAT32, TensorStorageType::SINGLE_TEXTURE_2D, Layout::HWDC}, in TEST_F()
204 {DataType::FLOAT32, TensorStorageType::SINGLE_TEXTURE_2D, Layout::HWDC}, in TEST_F()
241 {DataType::FLOAT16, TensorStorageType::SINGLE_TEXTURE_2D, Layout::HWDC}, in TEST_F()
245 {DataType::FLOAT16, TensorStorageType::SINGLE_TEXTURE_2D, Layout::HWDC}, in TEST_F()
|
| D | serialization.cc | 124 case Layout::HWDC: in ToFB()
125 return data::Layout::HWDC; in ToFB()
218 case data::Layout::HWDC: in ToEnum()
219 return Layout::HWDC; in ToEnum()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | shape.h | 59 HWDC = 11, enumerator
196 using HWDC = StrongShape<Layout::HWDC>; variable
366 TFLITE_GPU_LAYOUT_TRAITS(HWDC, Axis::HEIGHT, Axis::WIDTH, Axis::DEPTH,
599 case Layout::HWDC:
600 return f.template operator()<Layout::HWDC>();
|
| D | shape.cc | 92 case Layout::HWDC: in ToString()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/metal/ |
| D | metal_spatial_tensor_test.mm | 121 BHWDC(1, 6, 7, 4, 3), {data_type, storage_type, Layout::HWDC}, device));
123 BHWDC(1, 1, 4, 3, 12), {data_type, storage_type, Layout::HWDC}, device));
125 BHWDC(1, 6, 1, 7, 7), {data_type, storage_type, Layout::HWDC}, device));
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/task/ |
| D | serialization_base.fbs | 137 HWDC = 3,
|
| D | tensor_desc.cc | 720 } else if (layout == Layout::HWDC || in GetPhysicalCoords()
763 (layout == Layout::HWDC || layout == Layout::BHWDC)) { in HasAxis()
|
| D | serialization_base_generated.h | 274 HWDC = 3, enumerator
285 Layout::HWDC, in EnumValuesLayout()
|