Searched refs:SetBHWDCShape (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/ |
| D | cl_test.cc | 64 descriptor_with_shape.SetBHWDCShape(dst_cpu[i]->GetBHWDCShape()); in ExecuteGpuOperationInternal()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| D | gather_test_util.cc | 52 dst.SetBHWDCShape(BHWDC(1, 1, 9, 1, 1)); in GatherWidthIntTest()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/metal/kernels/ |
| D | test_util.cc | 70 descriptor_with_shape.SetBHWDCShape(dst_cpu[i]->GetBHWDCShape()); in ExecuteGpuOperationInternal()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/task/ |
| D | testing_util.cc | 98 dst_cpu_descs[i].SetBHWDCShape(dst_sizes[i]); in ExecuteGPUOperation()
|
| D | tensor_desc.h | 158 void SetBHWDCShape(const BHWDC& new_shape) { shape_ = new_shape; } in SetBHWDCShape() function
|
| D | tensor_desc.cc | 1608 tensor_desc.SetBHWDCShape(BHWDC(1, 1, 1, 1, src.shape.v)); in CreateConstantLinearTensorDescriptor()
1626 tensor_desc.SetBHWDCShape(BHWDC(1, height, width, 1, 4)); in CreateConstantHWVec4TensorDescriptor()
|
| D | serialization_base.cc | 424 desc->SetBHWDCShape(BHWDC(fb_desc->shape()->b(), fb_desc->shape()->h(), in Decode()
|