Searched refs:ConvertToPHWC4 (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | convert.h | 35 absl::Status ConvertToPHWC4(absl::Span<const float> in, const BHWC& shape,
50 std::vector<float> ConvertToPHWC4(
52 std::vector<float> ConvertToPHWC4(const Tensor<HWC, DataType::FLOAT32>& tensor);
|
| D | convert.cc | 234 absl::Status ConvertToPHWC4(absl::Span<const float> in, const BHWC& shape, in ConvertToPHWC4() function
367 std::vector<float> ConvertToPHWC4( in ConvertToPHWC4() function
370 ConvertToPHWC4(tensor.data, tensor.shape, in ConvertToPHWC4()
377 std::vector<float> ConvertToPHWC4( in ConvertToPHWC4() function
382 ConvertToPHWC4(tensor.data, batched_shape, in ConvertToPHWC4()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/ |
| D | prelu.cc | 112 MakeReadonlyObject(obj_size, ConvertToPHWC4(*alpha))}}, in GenerateCode()
132 MakeReadonlyObject(obj_size, ConvertToPHWC4(*alpha))}}, in GenerateCode()
|
| D | mul.cc | 130 ConvertToPHWC4( in GenerateMultiplyScalarCode()
|
| D | converter_test.cc | 111 ConvertToPHWC4(absl::MakeConstSpan(input.data(), input.size()), shape, in RunToTensorTest()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/gl/converters/ |
| D | bhwc_to_phwc4_test.cc | 49 ConvertToPHWC4(absl::MakeConstSpan(input.data(), input.size()), shape, in RunTest()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/gl/ |
| D | object_manager.cc | 31 ConvertToPHWC4(tensor.data, tensor.shape, absl::MakeSpan(transposed))); in CreatePHWC4BufferFromTensor()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/ |
| D | gl_delegate.cc | 116 return ConvertToPHWC4( in CopyToBufferHandle()
|