Searched refs:output_feature_dimension (Results 1 – 19 of 19) sorted by relevance
125 auto output_feature_dim = conv_dnums.output_feature_dimension(); in MatchBackwardFilter()474 int64 output_feature_dimension = dnums.kernel_output_feature_dimension(); in MatchBackwardInput() local479 CHECK_EQ(std::abs(input_feature_dimension - output_feature_dimension), 1LL); in MatchBackwardInput()481 int64 output_features = new_shape.dimensions(output_feature_dimension); in MatchBackwardInput()484 new_shape.set_dimensions(output_feature_dimension, in MatchBackwardInput()
89 output_layout.push_back(dnums.output_feature_dimension()); in StreamExecutorConvLayoutsToXlaLayouts()99 output_layout.push_back(dnums.output_feature_dimension()); in StreamExecutorConvLayoutsToXlaLayouts()
169 new_output_shape->dimensions(dnums.output_feature_dimension()); in PadForTensorCores()180 pad_dim(new_output_shape, dnums.output_feature_dimension()); in PadForTensorCores()
158 conv->convolution_dimension_numbers().output_feature_dimension()) { in FindConvWithRelu()211 conv->convolution_dimension_numbers().output_feature_dimension()); in TryRewriteToCudnnForwardRelu()
221 output_shape.dimensions(dnums.output_feature_dimension())) in RunCudnnConvImpl()278 output_shape.dimensions(dnums.output_feature_dimension())) in RunCudnnConvImpl()
220 int64 output_feature_dimension = dim_numbers.output_feature_dimension(); in HandleBatchGroupCount() local236 output_feature_dimension, batch_group_count, add); in HandleBatchGroupCount()490 auto output_feature_dim = dim_numbers.output_feature_dimension(); in HandleConvolution()
446 const int64 output_feature_dim = dnums.output_feature_dimension(); in HandleConvolution()
1658 output_dnums[1] = dnums.output_feature_dimension(); in InferConvolveShape()1784 dimensions[dnums.output_feature_dimension()] = kernel_output_features; in InferConvolveShape()
1078 const int64 output_z_dim = dnums.output_feature_dimension(); in HandleConvolution()
3070 output_dims[dnums.output_feature_dimension()] = 'f'; in ConvolutionDimensionNumbersToString()
3584 dnums.output_feature_dimension() || in SimplifyConvToDot()
107 dnums.output_feature_dimension() == in PotentiallyImplementedAsEigenConvolution()
103 auto output_feature_dim = dnums.output_feature_dimension(); in Run()
966 llvm::Value* output_feature = index[dnums.output_feature_dimension()]; in EmitElementalConvolution()
111 dim_nums.set_output_batch_dimension(dim_nums.output_feature_dimension()); in XLA_TEST_F()
1666 dimension_numbers.output_feature_dimension = 11717 dimension_numbers.output_feature_dimension = out_spec.index('C')1945 self.output_feature_dimension = 0
116 dnums.output_feature_dimension = 1
507 int64 output_feature_dimension = 10; field
2702 dnum.output_feature_dimension(), in Validate()2709 dnum.output_batch_dimension(), dnum.output_feature_dimension(), in Validate()