| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | convert.cc | 45 if (in.size() != shape.DimensionsProduct()) { in ConvertToPHWO4I4()
48 in.size(), " != ", shape.DimensionsProduct())); in ConvertToPHWO4I4()
124 if (in.size() != shape.DimensionsProduct()) { in ConvertToPHWO4I4()
127 in.size(), " != ", shape.DimensionsProduct())); in ConvertToPHWO4I4()
178 if (in.size() != shape.DimensionsProduct()) { in ConvertToPIOHW4()
181 in.size(), " != ", shape.DimensionsProduct())); in ConvertToPIOHW4()
220 if (in.size() != shape.DimensionsProduct()) { in ValidateConvertToPHWC4()
223 in.size(), " != ", shape.DimensionsProduct())); in ValidateConvertToPHWC4()
239 shape.DimensionsProduct() * sizeof(float)); in ConvertToPHWC4()
401 if (out.size() != shape.DimensionsProduct()) { in ValidateConvertFromPHWC4()
[all …]
|
| D | tensor.h | 89 shape.DimensionsProduct(), 0);
|
| D | shape.h | 128 int64_t DimensionsProduct() const { in DimensionsProduct() function
473 bool empty() const { return DimensionsProduct() == 0; }
485 int64_t DimensionsProduct() const {
|
| D | winograd_util.cc | 120 dst_weights->data.resize(dst_shape.DimensionsProduct()); in RearrangeWeightsToWinograd4x4To6x6Weights()
|
| D | model_builder.cc | 522 weights.data.resize(weights.shape.DimensionsProduct()); in TransposeWeights()
821 if (input->tensor.shape.DimensionsProduct() / batch_size != weights_width) { in Parse()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| D | conv_weights_converter_test_util.cc | 38 src_tensor.data.resize(src_shape.DimensionsProduct(), 2.0); in ConvolutionWeightsConverterTest()
116 weights.data.resize(weights_shape.DimensionsProduct()); in ConverterToConvWeights1x1OutX4Test()
150 weights.data.resize(weights_shape.DimensionsProduct()); in ConverterToConvWeights1x1OutX4UnalignedTest()
183 weights.data.resize(weights_shape.DimensionsProduct()); in ConverterToConvWeights1x1OutX2Test()
216 weights.data.resize(weights_shape.DimensionsProduct()); in ConverterToConvWeightsOutX2Test()
251 weights.data.resize(weights_shape.DimensionsProduct()); in ConverterToConvTransposedWeights4x4Test()
285 weights.data.resize(weights_shape.DimensionsProduct()); in ConverterToConvWeights4xTexturesTest()
|
| D | winograd_test_util.cc | 105 biases.data.resize(biases.shape.DimensionsProduct()); in Winograd36To4x4Tile4x1Test()
239 biases.data.resize(biases.shape.DimensionsProduct()); in Winograd36To4x4Test()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/gl/converters/ |
| D | phwc4_to_bhwc_test.cc | 48 std::vector<float> output(shape.DimensionsProduct(), 0); in RunTest()
63 shape.DimensionsProduct(), &output_buffer)); in RunTest()
|
| D | util.h | 42 return shape.DimensionsProduct() * sizeof(float); in BytesForBHWC()
|
| D | bhwc_to_phwc4_test.cc | 46 std::vector<float> input = GenerateFloats(0.01, shape.DimensionsProduct()); in RunTest()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/ |
| D | converter_test.cc | 52 std::vector<float> output(shape.DimensionsProduct(), 0); in RunFromTensorTest()
67 shape.DimensionsProduct(), &output_buffer)); in RunFromTensorTest()
108 std::vector<float> input = GenerateFloats(0.01, shape.DimensionsProduct()); in RunToTensorTest()
|
| D | test_util.cc | 64 inputs_[index].shape.DimensionsProduct() != data.size()) { in PopulateTensor()
123 tensor.data.reserve(output->tensor.shape.DimensionsProduct()); in Invoke()
|
| D | converter.cc | 86 return shape.DimensionsProduct() * sizeof(float); in SizeInBytesBHWC()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/ |
| D | tensor_test.cc | 37 tensor_cpu.data.resize(shape.DimensionsProduct()); in TensorGenericTest()
43 tensor_gpu.data.resize(shape.DimensionsProduct()); in TensorGenericTest()
66 tensor_cpu.data.resize(shape.DimensionsProduct()); in Tensor5DGenericTest()
72 tensor_gpu.data.resize(shape.DimensionsProduct()); in Tensor5DGenericTest()
|
| D | inference_context.cc | 727 result->data.resize(dst_shape.DimensionsProduct()); in GetOutputTensor()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/metal/ |
| D | metal_spatial_tensor_test.mm | 45 tensor_cpu.data.resize(shape.DimensionsProduct());
51 tensor_gpu.data.resize(shape.DimensionsProduct());
76 tensor_cpu.data.resize(shape.DimensionsProduct());
82 tensor_gpu.data.resize(shape.DimensionsProduct());
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/ |
| D | cl_test.cc | 94 dst_cpu[i]->data = std::vector<float>(dst_sizes[i].DimensionsProduct(), 0); in ExecuteGPUOperation()
142 dst_cpu[i]->data = std::vector<float>(dst_sizes[i].DimensionsProduct(), 0); in ExecuteGPUOperation()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/transformations/ |
| D | make_padding_test.cc | 56 std::vector<float>(const_attr.tensor.shape.DimensionsProduct(), 0); in TEST()
|
| D | fuse_mul_to_conv_test.cc | 51 conv_attr.weights.data.resize(conv_attr.weights.shape.DimensionsProduct()); in TEST()
108 conv_attr.weights.data.resize(conv_attr.weights.shape.DimensionsProduct()); in TEST()
|
| D | fuse_add_to_conv_test.cc | 51 conv_attr.weights.data.resize(conv_attr.weights.shape.DimensionsProduct()); in TEST()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/gl/ |
| D | object_manager.cc | 44 tensor->data.resize(tensor->shape.DimensionsProduct()); in CopyFromPHWC4Buffer()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/metal/kernels/ |
| D | test_util.cc | 113 dst_cpu[i]->data = std::vector<float>(dst_sizes[i].DimensionsProduct(), 0); in ExecuteGPUOperation()
|
| D | conv_test.mm | 232 attr.weights.data.resize(attr.weights.shape.DimensionsProduct());
237 attr.bias.data.resize(attr.bias.shape.DimensionsProduct());
254 src_tensor.data.resize(src_tensor.shape.DimensionsProduct());
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/ |
| D | metal_delegate.mm | 379 int bhwc_length = static_cast<int>(sizeof(float) * input_tensor.shape.DimensionsProduct());
416 int bhwc_length = static_cast<int>(sizeof(float) * output_tensor.shape.DimensionsProduct());
475 std::memcpy(gpu_ptr, tensor->data.f, input.shape.DimensionsProduct() * sizeof(float));
565 std::memcpy(tensor->data.f, gpu_ptr, output.shape.DimensionsProduct() * sizeof(float));
|