Searched refs:GetElementsSizeForPHWC4 (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/gpu/gl/converters/ |
| D | bhwc_to_phwc4_test.cc | 47 std::vector<float> output(GetElementsSizeForPHWC4(shape), 0); in RunTest()
62 GetElementsSizeForPHWC4(shape), &output_buffer)); in RunTest()
|
| D | phwc4_to_bhwc_test.cc | 47 GenerateFloats(0.01, GetElementsSizeForPHWC4(shape)); in RunTest()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/ |
| D | converter_test.cc | 51 GenerateFloats(0.01, GetElementsSizeForPHWC4(shape)); in RunFromTensorTest()
109 std::vector<float> output(GetElementsSizeForPHWC4(shape), 0); in RunToTensorTest()
124 GetElementsSizeForPHWC4(shape), &output_buffer)); in RunToTensorTest()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | convert.cc | 225 if (out.size() != GetElementsSizeForPHWC4(shape)) { in ValidateConvertToPHWC4()
228 out.size(), " != ", GetElementsSizeForPHWC4(shape))); in ValidateConvertToPHWC4()
369 std::vector<float> transposed(GetElementsSizeForPHWC4(tensor.shape)); in ConvertToPHWC4()
381 std::vector<float> transposed(GetElementsSizeForPHWC4(batched_shape)); in ConvertToPHWC4()
389 uint32_t GetElementsSizeForPHWC4(const BHWC& shape) { in GetElementsSizeForPHWC4() function
396 if (in.size() != GetElementsSizeForPHWC4(shape)) { in ValidateConvertFromPHWC4()
399 in.size(), " != ", GetElementsSizeForPHWC4(shape))); in ValidateConvertFromPHWC4()
|
| D | convert.h | 41 uint32_t GetElementsSizeForPHWC4(const BHWC& shape);
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/gl/ |
| D | object_manager.cc | 29 std::vector<float> transposed(GetElementsSizeForPHWC4(tensor.shape)); in CreatePHWC4BufferFromTensor()
38 GetElementsSizeForPHWC4(tensor_ref.shape), gl_buffer); in CreatePHWC4BufferFromTensorRef()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/ |
| D | gl_delegate.cc | 200 GetElementsSizeForPHWC4(input->tensor.shape), &buffer)); in Prepare()
238 GetElementsSizeForPHWC4(output->tensor.shape), &buffer)); in Prepare()
|
| D | metal_delegate.mm | 381 static_cast<int>(storage_type_size * GetElementsSizeForPHWC4(input_tensor.shape));
418 static_cast<int>(storage_type_size * GetElementsSizeForPHWC4(output_tensor.shape));
|