Home
last modified time | relevance | path

Searched refs:output_elements (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/lite/micro/kernels/arc_mli/
Ddepthwise_conv_slicing_test.cc206 const int output_elements = 20; in TF_LITE_MICRO_TEST() local
222 int8_t golden_quantized[output_elements]; in TF_LITE_MICRO_TEST()
240 const int output_elements = 20; in TF_LITE_MICRO_TEST() local
265 int8_t golden_quantized[output_elements]; in TF_LITE_MICRO_TEST()
292 const int output_elements = 80; in TF_LITE_MICRO_TEST() local
312 int8_t golden_quantized[output_elements]; in TF_LITE_MICRO_TEST()
330 const int output_elements = 80; in TF_LITE_MICRO_TEST() local
362 int8_t golden_quantized[output_elements]; in TF_LITE_MICRO_TEST()
389 const int output_elements = 4; in TF_LITE_MICRO_TEST() local
404 int8_t golden_quantized[output_elements]; in TF_LITE_MICRO_TEST()
[all …]
/external/tensorflow/tensorflow/lite/micro/kernels/
Ddepthwise_conv_test.cc278 const int output_elements = 8; in TF_LITE_MICRO_TEST() local
295 uint8_t golden_quantized[output_elements]; in TF_LITE_MICRO_TEST()
296 uint8_t output_data[output_elements]; in TF_LITE_MICRO_TEST()
324 const int output_elements = 24; in TF_LITE_MICRO_TEST() local
342 uint8_t golden_quantized[output_elements]; in TF_LITE_MICRO_TEST()
343 uint8_t output_data[output_elements]; in TF_LITE_MICRO_TEST()
391 const int output_elements = 8; in TF_LITE_MICRO_TEST() local
406 uint8_t golden_quantized[output_elements]; in TF_LITE_MICRO_TEST()
407 uint8_t output_data[output_elements]; in TF_LITE_MICRO_TEST()
475 const int output_elements = 8; in TF_LITE_MICRO_TEST() local
[all …]
Dconv_test.cc202 const int output_elements = 24; in TF_LITE_MICRO_TEST() local
210 uint8_t golden_quantized[output_elements]; in TF_LITE_MICRO_TEST()
273 const int output_elements = 24; in TF_LITE_MICRO_TEST() local
281 int8_t golden_quantized[output_elements]; in TF_LITE_MICRO_TEST()
356 constexpr int output_elements = 4 * 3; in TF_LITE_MICRO_TEST() local
357 int8_t output_data[output_elements]; in TF_LITE_MICRO_TEST()
359 const float golden_data[output_elements] = {11, 2, 3, 21, 2, 3, in TF_LITE_MICRO_TEST()
370 int8_t golden_quantized[output_elements]; in TF_LITE_MICRO_TEST()
407 constexpr int output_elements = 4 * 3; in TF_LITE_MICRO_TEST() local
408 int8_t output_data[output_elements]; in TF_LITE_MICRO_TEST()
[all …]
/external/XNNPACK/bench/
Df32-conv-hwc2chw.cc62 const size_t output_elements = output_height * output_width * output_channels; in DConvHWC2CHW3X3S2P1Benchmark() local
65 sizeof(float) * (weights_elements + output_elements)); in DConvHWC2CHW3X3S2P1Benchmark()
79 std::vector<float> output(output_elements * num_buffers); in DConvHWC2CHW3X3S2P1Benchmark()
97 output.data() + buffer_index * output_elements, in DConvHWC2CHW3X3S2P1Benchmark()
Df32-conv-hwc.cc60 const size_t output_elements = output_height * output_width * output_channels; in DConv3X3S2P1Benchmark() local
63 sizeof(float) * (weights_elements + output_elements)); in DConv3X3S2P1Benchmark()
77 std::vector<float> output(output_elements * num_buffers); in DConv3X3S2P1Benchmark()
95 output.data() + buffer_index * output_elements, in DConv3X3S2P1Benchmark()
Ddeconvolution.cc64 const size_t output_elements = batch_size * output_height * output_width * output_pixel_stride; in xnnpack_deconvolution_qu8() local
74 sizeof(float) * (kernel.size() + bias.size() + output_elements)); in xnnpack_deconvolution_qu8()
75 std::vector<uint8_t> output(output_elements * num_buffers); in xnnpack_deconvolution_qu8()
102 input.data(), output.data() + i * output_elements, in xnnpack_deconvolution_qu8()
182 const size_t output_elements = batch_size * output_height * output_width * output_pixel_stride; in xnnpack_deconvolution_f32() local
192 sizeof(float) * (kernel.size() + bias.size() + output_elements)); in xnnpack_deconvolution_f32()
193 std::vector<float> output(output_elements * num_buffers); in xnnpack_deconvolution_f32()
219 input.data(), output.data() + i * output_elements, in xnnpack_deconvolution_f32()
Dconvolution.cc77 const size_t output_elements = batch_size * output_height * output_width * output_pixel_stride; in xnnpack_convolution_qu8() local
87 …zeof(uint8_t) * kernel.size() + sizeof(int32_t) * bias.size() + sizeof(uint8_t) * output_elements); in xnnpack_convolution_qu8()
88 std::vector<uint8_t> output(output_elements * num_buffers); in xnnpack_convolution_qu8()
114 input.data(), output.data() + i * output_elements, in xnnpack_convolution_qu8()
198 const size_t output_elements = batch_size * output_height * output_width * output_pixel_stride; in xnnpack_convolution_qs8() local
208 …sizeof(int8_t) * kernel.size() + sizeof(int32_t) * bias.size() + sizeof(int8_t) * output_elements); in xnnpack_convolution_qs8()
209 std::vector<int8_t> output(output_elements * num_buffers); in xnnpack_convolution_qs8()
234 input.data(), output.data() + i * output_elements, in xnnpack_convolution_qs8()
320 const size_t output_elements = batch_size * output_height * output_width * output_pixel_stride; in xnnpack_convolution_f16() local
330 sizeof(uint16_t) * (kernel.size() + bias.size() + output_elements)); in xnnpack_convolution_f16()
[all …]
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/sparse_ops/
Dfully_connected.h41 const int output_elements = output_shape.FlatSize(); in FullyConnectedSparseWeight() local
52 for (int i = 0; i < output_elements; ++i) { in FullyConnectedSparseWeight()
171 const int output_elements = output_shape.FlatSize(); in FullyConnectedSparseWeight1x4() local
172 memset(output_data, 0, output_elements * sizeof(float)); in FullyConnectedSparseWeight1x4()
/external/tensorflow/tensorflow/compiler/xla/service/
Dbfloat16_normalization.cc348 std::vector<HloInstruction*> output_elements(hlo->operand_count()); in HandleMultipleOutputs() local
353 output_elements[i] = computation_->AddInstruction( in HandleMultipleOutputs()
363 output_elements[i] = in HandleMultipleOutputs()
367 HloInstruction::CreateTuple(output_elements)); in HandleMultipleOutputs()
/external/tensorflow/tensorflow/core/grappler/costs/
Dop_level_cost_estimator.cc2523 const int64 output_elements = CalculateTensorElementCount( in PredictResizeBilinear() local
2599 ops += (add_cost * 3 + sub_cost_float * 3 + mul_cost * 3) * output_elements; in PredictResizeBilinear()
2627 const int64 output_elements = CalculateTensorElementCount( in PredictCropAndResize() local
2664 output_elements; in PredictCropAndResize()
2669 ops += cast_to_float_cost * output_elements; in PredictCropAndResize()
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc3229 int output_elements = result_ty.getNumElements(); in rewriteWithUnknownBegin() local
3242 if (input_elements_sliced != output_elements) { in rewriteWithUnknownBegin()