Home
last modified time | relevance | path

Searched defs:output_elements (Results 1 – 14 of 14) sorted by relevance

/aosp_15_r20/external/XNNPACK/bench/
H A Dconvolution.cc70 const size_t output_elements = batch_size * output_height * output_width * output_pixel_stride; in xnnpack_convolution_qu8() local
191 const size_t output_elements = batch_size * output_height * output_width * output_pixel_stride; in xnnpack_convolution_qs8() local
313 const size_t output_elements = batch_size * output_height * output_width * output_pixel_stride; in xnnpack_convolution_f16() local
428 const size_t output_elements = batch_size * output_height * output_width * output_pixel_stride; in xnnpack_convolution_f32() local
H A Ddeconvolution.cc66 const size_t output_elements = batch_size * output_height * output_width * output_channels; in xnnpack_deconvolution_qu8() local
185 const size_t output_elements = batch_size * output_height * output_width * output_channels; in xnnpack_deconvolution_f32() local
H A Df16-conv-hwc2chw.cc66 const size_t output_elements = output_height * output_width * output_channels; in f16_conv_hwc2chw() local
H A Df32-conv-hwc2chw.cc64 const size_t output_elements = output_height * output_width * output_channels; in f32_conv_hwc2chw() local
H A Df32-conv-hwc.cc62 const size_t output_elements = output_height * output_width * output_channels; in f32_conv_hwc() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/sparse_ops/
H A Dfully_connected.h44 const int output_elements = output_shape.FlatSize(); in FullyConnectedSparseWeight() local
206 const int output_elements = output_shape.FlatSize(); in FullyConnectedSparseWeight1x16() local
230 const int output_elements = output_shape.FlatSize(); in FullyConnectedSparseWeight1x4() local
/aosp_15_r20/external/pytorch/test/custom_backend/
H A Dtest_custom_backend.cpp17 auto output_elements = output.toTupleRef().elements(); in load_serialized_lowered_module_and_execute() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dedit_distance_op.cc168 const auto output_elements = output_shape.num_elements(); in Compute() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dbfloat16_normalization.cc347 std::vector<HloInstruction*> output_elements(hlo->operand_count()); in HandleMultipleOutputs() local
H A Dwhile_loop_concat_code_motion.cc758 std::vector<HloInstruction*> output_elements( in RewriteLoopWithConcatGroups() local
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/costs/
H A Dop_level_cost_estimator.cc2570 const int64_t output_elements = CalculateTensorElementCount( in PredictResizeBilinear() local
2675 const int64_t output_elements = CalculateTensorElementCount( in PredictCropAndResize() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dfully_connected.cc924 const int output_elements = output_shape.FlatSize(); in VerifySparsity() local
/aosp_15_r20/external/federated-compute/fcp/client/
H A Dfl_runner.cc352 auto output_elements = output_tensors[0].NumElements(); in ParseEligibilityEvalPlanOutput() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
H A Dlegalize_tf.cc3710 int output_elements = result_ty.getNumElements(); in rewriteWithUnknownBegin() local