| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | activations_test.cc | 290 const int elements_count = 10; in TF_LITE_MICRO_TEST() local
294 uint8_t input_quantized[elements_count]; in TF_LITE_MICRO_TEST()
297 uint8_t golden_quantized[elements_count]; in TF_LITE_MICRO_TEST()
298 uint8_t output_data[elements_count]; in TF_LITE_MICRO_TEST()
312 const int elements_count = 10; in TF_LITE_MICRO_TEST() local
316 uint8_t input_quantized[elements_count]; in TF_LITE_MICRO_TEST()
319 uint8_t golden_quantized[elements_count]; in TF_LITE_MICRO_TEST()
320 uint8_t output_data[elements_count]; in TF_LITE_MICRO_TEST()
334 const int elements_count = 10; in TF_LITE_MICRO_TEST() local
338 int8_t input_quantized[elements_count]; in TF_LITE_MICRO_TEST()
[all …]
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| D | depthwise_conv.h | 75 const int elements_count = kernel_x * kernel_y * dst_slices; in UploadWeightsForDWConv2D() local
80 std::vector<uint8_t> data(float4_size * elements_count); in UploadWeightsForDWConv2D()
84 RearrangeWeightsForDWConv2D(weights, absl::MakeSpan(ptr, elements_count)); in UploadWeightsForDWConv2D()
87 RearrangeWeightsForDWConv2D(weights, absl::MakeSpan(ptr, elements_count)); in UploadWeightsForDWConv2D()
94 desc.size = float4_size * elements_count; in UploadWeightsForDWConv2D()
150 const int elements_count = kernel_x * kernel_y * kernel_z * dst_slices; in UploadWeightsForDWConv3D() local
155 std::vector<uint8_t> data(float4_size * elements_count); in UploadWeightsForDWConv3D()
159 RearrangeWeightsForDWConv3D(weights, absl::MakeSpan(ptr, elements_count)); in UploadWeightsForDWConv3D()
162 RearrangeWeightsForDWConv3D(weights, absl::MakeSpan(ptr, elements_count)); in UploadWeightsForDWConv3D()
169 desc.size = float4_size * elements_count; in UploadWeightsForDWConv3D()
|
| D | convolution_transposed.h | 150 const int elements_count = in UploadWeights() local
155 std::vector<uint8_t> data(float4_size * elements_count); in UploadWeights()
161 absl::MakeSpan(ptr, elements_count)); in UploadWeights()
164 absl::MakeSpan(ptr, elements_count)); in UploadWeights()
170 absl::MakeSpan(ptr, elements_count)); in UploadWeights()
173 absl::MakeSpan(ptr, elements_count)); in UploadWeights()
181 desc.size = float4_size * elements_count; in UploadWeights()
|
| D | depthwise_conv_3x3_stride_h2.h | 77 const int elements_count = texture_width * texture_height; in UploadWeightsAndBiases() local
81 std::vector<uint8_t> data(float4_size * elements_count); in UploadWeightsAndBiases()
85 absl::MakeSpan(ptr, elements_count)); in UploadWeightsAndBiases()
89 absl::MakeSpan(ptr, elements_count)); in UploadWeightsAndBiases()
96 desc.size = float4_size * elements_count; in UploadWeightsAndBiases()
|
| D | depthwise_conv_3x3.h | 83 const int elements_count = texture_width * texture_height; in UploadWeightsAndBiases() local
87 std::vector<uint8_t> data(float4_size * elements_count); in UploadWeightsAndBiases()
91 absl::MakeSpan(ptr, elements_count)); in UploadWeightsAndBiases()
95 absl::MakeSpan(ptr, elements_count)); in UploadWeightsAndBiases()
102 desc.size = float4_size * elements_count; in UploadWeightsAndBiases()
|
| D | conv_powervr.h | 305 const int elements_count = weights.shape.d * weights.shape.h * in UploadWeights() local
311 std::vector<uint8_t> data(float4_size * elements_count); in UploadWeights()
317 absl::MakeSpan(ptr, elements_count)); in UploadWeights()
320 absl::MakeSpan(ptr, elements_count)); in UploadWeights()
326 absl::MakeSpan(ptr, elements_count)); in UploadWeights()
329 absl::MakeSpan(ptr, elements_count)); in UploadWeights()
337 desc.size = float4_size * elements_count; in UploadWeights()
|
| D | conv_buffer_1x1.h | 134 const int elements_count = in UploadWeights() local
141 desc.size = float4_size * elements_count; in UploadWeights()
147 absl::MakeSpan(ptr, elements_count)); in UploadWeights()
151 absl::MakeSpan(ptr, elements_count)); in UploadWeights()
|
| D | fully_connected.h | 153 const int elements_count = src_depth * dst_depth * 4; in UploadWeights() local
162 desc.size = float4_size * elements_count; in UploadWeights()
179 desc.data.resize(float4_size * elements_count); in UploadWeights()
|
| /external/gemmlowp/meta/generators/ |
| D | streams_common.py | 57 def _GenerateLoadAggregateStore(emitter, registers, lanes_count, elements_count, argument
62 elements_count))
66 if elements_count is not 8:
70 emitter.EmitVLoadE(8, elements_count, row, input_address, None)
206 elements_count, aggregators, input_address, argument
211 (lanes_count, elements_count))
215 if elements_count is not 8:
218 block = emitter.EmitLoadColBlock(registers, 8, lanes_count, elements_count,
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/metal/ |
| D | buffer.h | 91 const int elements_count = size_ / sizeof(T); in ReadData() local
92 result->resize(elements_count); in ReadData()
|
| D | metal_spatial_tensor.cc | 386 const int elements_count = in WriteDataBHWDC() local
389 const size_t data_size = elements_count * SizeOf(descriptor_.data_type); in WriteDataBHWDC()
393 data_f.reset(new float[elements_count]); in WriteDataBHWDC()
397 data_h.reset(new half[elements_count]); in WriteDataBHWDC()
452 const int elements_count = in ReadDataBHWDC() local
454 const size_t data_size = elements_count * SizeOf(descriptor_.data_type); in ReadDataBHWDC()
458 data_f.reset(new float[elements_count]); in ReadDataBHWDC()
461 data_h.reset(new half[elements_count]); in ReadDataBHWDC()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/special/ |
| D | fc_fc_add.h | 130 const int elements_count = src_depth * dst_depth * 4; in UploadWeights() local
139 desc.size = float4_size * elements_count; in UploadWeights()
159 desc.data.resize(float4_size * elements_count); in UploadWeights()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/ |
| D | buffer.h | 102 const int elements_count = size_ / sizeof(T); in ReadData() local
103 result->resize(elements_count); in ReadData()
|
| D | texture2d.h | 119 const int elements_count = width_ * height_ * 4; in ReadData() local
120 result->resize(elements_count); in ReadData()
|
| D | tensor.cc | 490 const int elements_count = in WriteDataBHWDC() local
493 const size_t data_size = elements_count * SizeOf(descriptor_.data_type); in WriteDataBHWDC()
497 data_f.reset(new float[elements_count]); in WriteDataBHWDC()
501 data_h.reset(new half[elements_count]); in WriteDataBHWDC()
552 const int elements_count = in ReadDataBHWDC() local
554 const size_t data_size = elements_count * SizeOf(descriptor_.data_type); in ReadDataBHWDC()
558 data_f.reset(new float[elements_count]); in ReadDataBHWDC()
561 data_h.reset(new half[elements_count]); in ReadDataBHWDC()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/task/ |
| D | tensor_desc.cc | 908 int elements_count = shape.b * shape.w * shape.h * shape.d * aligned_channels; in UploadData() local
909 data.resize(elements_count * SizeOf(data_type)); in UploadData()
|