Searched refs:input_batch_dim (Results 1 – 5 of 5) sorted by relevance
40 auto input_batch_dim = dnums.input_batch_dimension(); in Run() local61 new_input_dim_order[0] = input_batch_dim; in Run()62 new_input_dims[0] = input->shape().dimensions(input_batch_dim); in Run()
118 auto input_batch_dim = conv_dnums.input_batch_dimension(); in MatchBackwardFilter() local151 if (input_batch_dim == output_batch_dim && in MatchBackwardFilter()231 backward_conv_dnums.set_input_feature_dimension(input_batch_dim); in MatchBackwardFilter()
444 const int64 input_batch_dim = dnums.input_batch_dimension(); in HandleConvolution() local451 const int64 batch = ShapeUtil::GetDimension(lhs_shape, input_batch_dim); in HandleConvolution()
1071 const int64 input_batch_dim = dnums.input_batch_dimension(); in HandleConvolution() local1084 ShapeUtil::GetDimension(lhs_shape, input_batch_dim); in HandleConvolution()1166 lhs_dim_multipliers[input_batch_dim]; in HandleConvolution()1176 lhs_dim_multipliers[input_batch_dim]; in HandleConvolution()
2127 const int input_batch_dim = input_tensor.batch_size(); in ConvertReshape() local2140 if (input_batch_dim > 0) { // Batch size is fixed. in ConvertReshape()2146 } else if (reshape_batch_dim != input_batch_dim) { in ConvertReshape()2157 VLOG(1) << "input_batch_dim=" << input_batch_dim in ConvertReshape()