Home
last modified time | relevance | path

Searched defs: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
240 const int output_elements = 20; in TF_LITE_MICRO_TEST() local
292 const int output_elements = 80; in TF_LITE_MICRO_TEST() local
330 const int output_elements = 80; in TF_LITE_MICRO_TEST() local
389 const int output_elements = 4; in TF_LITE_MICRO_TEST() local
422 const int output_elements = 4; in TF_LITE_MICRO_TEST() local
478 const int output_elements = 8; in TF_LITE_MICRO_TEST() local
511 const int output_elements = 8; in TF_LITE_MICRO_TEST() local
/external/tensorflow/tensorflow/lite/micro/kernels/
Ddepthwise_conv_test.cc278 const int output_elements = 8; in TF_LITE_MICRO_TEST() local
324 const int output_elements = 24; in TF_LITE_MICRO_TEST() local
391 const int output_elements = 8; in TF_LITE_MICRO_TEST() local
475 const int output_elements = 8; in TF_LITE_MICRO_TEST() local
515 const int output_elements = 4; in TF_LITE_MICRO_TEST() local
560 const int output_elements = 8; in TF_LITE_MICRO_TEST() local
603 const int output_elements = 24; in TF_LITE_MICRO_TEST() local
690 const int output_elements = 8; in TF_LITE_MICRO_TEST() local
870 const int output_elements = 32; in TF_LITE_MICRO_TEST() local
Dconv_test.cc202 const int output_elements = 24; in TF_LITE_MICRO_TEST() local
273 const int output_elements = 24; in TF_LITE_MICRO_TEST() local
356 constexpr int output_elements = 4 * 3; in TF_LITE_MICRO_TEST() local
407 constexpr int output_elements = 4 * 3; in TF_LITE_MICRO_TEST() local
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/sparse_ops/
Dfully_connected.h41 const int output_elements = output_shape.FlatSize(); in FullyConnectedSparseWeight() local
171 const int output_elements = output_shape.FlatSize(); in FullyConnectedSparseWeight1x4() local
/external/XNNPACK/bench/
Df32-conv-hwc2chw.cc62 const size_t output_elements = output_height * output_width * output_channels; in DConvHWC2CHW3X3S2P1Benchmark() local
Ddeconvolution.cc64 const size_t output_elements = batch_size * output_height * output_width * output_pixel_stride; in xnnpack_deconvolution_qu8() local
182 const size_t output_elements = batch_size * output_height * output_width * output_pixel_stride; in xnnpack_deconvolution_f32() local
Df32-conv-hwc.cc60 const size_t output_elements = output_height * output_width * output_channels; in DConv3X3S2P1Benchmark() local
Dconvolution.cc77 const size_t output_elements = batch_size * output_height * output_width * output_pixel_stride; in xnnpack_convolution_qu8() local
198 const size_t output_elements = batch_size * output_height * output_width * output_pixel_stride; in xnnpack_convolution_qs8() local
320 const size_t output_elements = batch_size * output_height * output_width * output_pixel_stride; in xnnpack_convolution_f16() local
435 const size_t output_elements = batch_size * output_height * output_width * output_pixel_stride; in xnnpack_convolution_f32() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dbfloat16_normalization.cc348 std::vector<HloInstruction*> output_elements(hlo->operand_count()); in HandleMultipleOutputs() local
/external/tensorflow/tensorflow/core/grappler/costs/
Dop_level_cost_estimator.cc2523 const int64 output_elements = CalculateTensorElementCount( in PredictResizeBilinear() local
2627 const int64 output_elements = CalculateTensorElementCount( in PredictCropAndResize() local
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc3229 int output_elements = result_ty.getNumElements(); in rewriteWithUnknownBegin() local