Searched refs:input_batch_dimension (Results 1 – 21 of 21) sorted by relevance
44 input_layout.push_back(dnums.input_batch_dimension()); in StreamExecutorConvLayoutsToXlaLayouts()51 input_layout.push_back(dnums.input_batch_dimension()); in StreamExecutorConvLayoutsToXlaLayouts()
118 auto input_batch_dim = conv_dnums.input_batch_dimension(); in MatchBackwardFilter()
185 .set_count(input_shape.dimensions(dnums.input_batch_dimension())); in RunCudnnConvImpl()
384 new_conv->convolution_dimension_numbers().input_batch_dimension()); in TEST_F()386 dnums.input_batch_dimension(), in TEST_F()451 new_conv->convolution_dimension_numbers().input_batch_dimension()); in TEST_F()453 dnums.input_batch_dimension(), in TEST_F()
218 int64 input_batch_dimension = dim_numbers.input_batch_dimension(); in HandleBatchGroupCount() local222 int64 input_batch = activation->shape().dimensions(input_batch_dimension); in HandleBatchGroupCount()
98 inst->convolution_dimension_numbers().input_batch_dimension() == in ShouldSkipPadOnOperand()
142 transpose_dimensions[dnums.input_batch_dimension()]); in FoldTransposeIntoConvolution()
271 if (dimension == dimension_numbers.input_batch_dimension()) { in HandleConvolution()
444 const int64 input_batch_dim = dnums.input_batch_dimension(); in HandleConvolution()
1643 input_dnums[0] = dnums.input_batch_dimension(); in InferConvolveShape()1699 const int64 input_batch = lhs.dimensions(dnums.input_batch_dimension()); in InferConvolveShape()1792 if (i == dnums.input_batch_dimension()) { in InferConvolveShape()
1071 const int64 input_batch_dim = dnums.input_batch_dimension(); in HandleConvolution()
3055 lhs_dims[dnums.input_batch_dimension()] = 'b'; in ConvolutionDimensionNumbersToString()
3415 {dnums.input_batch_dimension(), dnums.input_feature_dimension()}) { in FoldConvInputPad()
104 return dnums.input_batch_dimension() == 0 && in PotentiallyImplementedAsEigenConvolution()
40 auto input_batch_dim = dnums.input_batch_dimension(); in Run()
1063 input_multi_index[dnums.input_batch_dimension()] = batch; in EmitElementalConvolution()1125 int64 input_batch = input_shape.dimensions(dnums.input_batch_dimension()); in HandleConvolution()
108 int64 old_input_batch_dim = dim_nums.input_batch_dimension(); in XLA_TEST_F()
1663 dimension_numbers.input_batch_dimension = 01714 dimension_numbers.input_batch_dimension = lhs_spec.index('N')1938 self.input_batch_dimension = 0
113 dnums.input_batch_dimension = 0
481 int64 input_batch_dimension = 7; field
2680 {dnum.input_batch_dimension(), dnum.input_feature_dimension(), in Validate()2686 dnum.input_batch_dimension(), dnum.input_feature_dimension(), in Validate()