Searched refs:output_batch_dim (Results 1 – 6 of 6) sorted by relevance
103 auto output_batch_dim = dnums.output_batch_dimension(); in Run() local105 new_output_dim_order[0] = output_batch_dim; in Run()106 new_conv_dims[0] = hlo->shape().dimensions(output_batch_dim); in Run()
110 TfLiteIntArray* output_batch_dim = in ApplyDelegate() local115 output_batch_dim->data[i] = output_batch_dimensions[i]; in ApplyDelegate()120 options.output_batch_dimensions = output_batch_dim; in ApplyDelegate()
187 auto output_batch_dim = conv_dnums.output_batch_dimension(); in MatchBackwardFilter() local323 backward_conv_dnums.set_kernel_input_feature_dimension(output_batch_dim); in MatchBackwardFilter()
2796 const int64 output_batch_dim = new_dim_numbers.output_batch_dimension(); in PerformSpaceToBatchOnConvolution() local2797 VLOG(1) << "output_batch_dim " << output_batch_dim in PerformSpaceToBatchOnConvolution()2805 output_batch_dim, output_split_spatial_dim, in PerformSpaceToBatchOnConvolution()
1136 const int64 output_batch_dim = dnums.output_batch_dimension();1226 lhs_linear_index += out_index[output_batch_dim] *
2591 const int output_batch_dim = output_shape_dims[0]; in ConvertReshape() local2602 << "\nresult_batch_dim=" << output_batch_dim in ConvertReshape()2608 if (input_batch_dim != -1 && output_batch_dim != -1) { in ConvertReshape()2609 reshape_may_change_batch_dim = (input_batch_dim != output_batch_dim); in ConvertReshape()2620 "; output_batch_dim=", output_batch_dim, ", ", in ConvertReshape()