Searched defs:output_elements (Results 1 – 12 of 12) sorted by relevance
44 const int output_elements = output_shape.FlatSize(); in FullyConnectedSparseWeight() local206 const int output_elements = output_shape.FlatSize(); in FullyConnectedSparseWeight1x16() local230 const int output_elements = output_shape.FlatSize(); in FullyConnectedSparseWeight1x4() local
66 const size_t output_elements = output_height * output_width * output_channels; in f16_conv_hwc2chw() local
70 const size_t output_elements = batch_size * output_height * output_width * output_pixel_stride; in xnnpack_convolution_qu8() local191 const size_t output_elements = batch_size * output_height * output_width * output_pixel_stride; in xnnpack_convolution_qs8() local313 const size_t output_elements = batch_size * output_height * output_width * output_pixel_stride; in xnnpack_convolution_f16() local428 const size_t output_elements = batch_size * output_height * output_width * output_pixel_stride; in xnnpack_convolution_f32() local
64 const size_t output_elements = output_height * output_width * output_channels; in f32_conv_hwc2chw() local
66 const size_t output_elements = batch_size * output_height * output_width * output_channels; in xnnpack_deconvolution_qu8() local185 const size_t output_elements = batch_size * output_height * output_width * output_channels; in xnnpack_deconvolution_f32() local
62 const size_t output_elements = output_height * output_width * output_channels; in f32_conv_hwc() local
168 const auto output_elements = output_shape.num_elements(); in Compute() local
347 std::vector<HloInstruction*> output_elements(hlo->operand_count()); in HandleMultipleOutputs() local
758 std::vector<HloInstruction*> output_elements( in RewriteLoopWithConcatGroups() local
2570 const int64_t output_elements = CalculateTensorElementCount( in PredictResizeBilinear() local2675 const int64_t output_elements = CalculateTensorElementCount( in PredictCropAndResize() local
924 const int output_elements = output_shape.FlatSize(); in VerifySparsity() local
3710 int output_elements = result_ty.getNumElements(); in rewriteWithUnknownBegin() local