Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/micro/kernels/
Ddepthwise_conv_test.cc264 const int output_elements = 8; in TF_LITE_MICRO_TEST() local
287 const int output_elements = 8; in TF_LITE_MICRO_TEST() local
305 uint8_t golden_quantized[output_elements]; in TF_LITE_MICRO_TEST()
306 uint8_t output_data[output_elements]; in TF_LITE_MICRO_TEST()
326 const int output_elements = 8; in TF_LITE_MICRO_TEST() local
348 const int output_elements = 8; in TF_LITE_MICRO_TEST() local
364 uint8_t golden_quantized[output_elements]; in TF_LITE_MICRO_TEST()
365 uint8_t output_data[output_elements]; in TF_LITE_MICRO_TEST()
423 const int output_elements = 8; in TF_LITE_MICRO_TEST() local
440 int8_t golden_quantized[output_elements]; in TF_LITE_MICRO_TEST()
[all …]
/external/XNNPACK/bench/
Df32-conv-hwc.cc61 const size_t output_elements = output_height * output_width * output_channels; in DConv3X3S2P1Benchmark() local
64 sizeof(float) * (weights_elements + output_elements)); in DConv3X3S2P1Benchmark()
78 std::vector<float> output(output_elements * num_buffers); in DConv3X3S2P1Benchmark()
96 output.data() + buffer_index * output_elements, in DConv3X3S2P1Benchmark()
Df32-conv-hwc2spchw.cc61 const size_t output_elements = output_height * output_width * output_channels; in DConvHWC2SpCHW3X3S2P1Benchmark() local
64 sizeof(float) * (weights_elements + output_elements)); in DConvHWC2SpCHW3X3S2P1Benchmark()
78 std::vector<float> output(output_elements * num_buffers); in DConvHWC2SpCHW3X3S2P1Benchmark()
96 output.data() + buffer_index * output_elements, in DConvHWC2SpCHW3X3S2P1Benchmark()
Ddeconvolution.cc65 const size_t output_elements = batch_size * output_height * output_width * output_pixel_stride; in xnnpack_deconvolution_q8() local
79 sizeof(float) * (kernel.size() + bias.size() + output_elements)); in xnnpack_deconvolution_q8()
80 std::vector<uint8_t> output(output_elements * num_buffers); in xnnpack_deconvolution_q8()
107 input.data(), output.data() + i * output_elements, in xnnpack_deconvolution_q8()
182 const size_t output_elements = batch_size * output_height * output_width * output_pixel_stride; in xnnpack_deconvolution_f32() local
196 sizeof(float) * (kernel.size() + bias.size() + output_elements)); in xnnpack_deconvolution_f32()
197 std::vector<float> output(output_elements * num_buffers); in xnnpack_deconvolution_f32()
223 input.data(), output.data() + i * output_elements, in xnnpack_deconvolution_f32()
Dconvolution.cc76 const size_t output_elements = batch_size * output_height * output_width * output_pixel_stride; in xnnpack_convolution_q8() local
90 …zeof(uint8_t) * kernel.size() + sizeof(int32_t) * bias.size() + sizeof(uint8_t) * output_elements); in xnnpack_convolution_q8()
91 std::vector<uint8_t> output(output_elements * num_buffers); in xnnpack_convolution_q8()
117 input.data(), output.data() + i * output_elements, in xnnpack_convolution_q8()
193 const size_t output_elements = batch_size * output_height * output_width * output_pixel_stride; in xnnpack_convolution_f32() local
207 sizeof(float) * (kernel.size() + bias.size() + output_elements)); in xnnpack_convolution_f32()
208 std::vector<float> output(output_elements * num_buffers); in xnnpack_convolution_f32()
232 input.data(), output.data() + i * output_elements, in xnnpack_convolution_f32()
/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/compiler/tf2xla/kernels/
Dstrided_slice_op.cc124 int64 output_elements = final_shape.num_elements(); in Compile() local
135 ctx, output_elements == input_elements_sliced, in Compile()
137 "The number of output elements ", output_elements, in Compile()