Home
last modified time | relevance | path

Searched refs:kernel_input_feature_dimension (Results 1 – 25 of 41) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/xla/service/
Dconvolution_group_converter.cc223 const int64 kernel_input_feature_dimension = in HandleBatchGroupCount() local
224 dim_numbers.kernel_input_feature_dimension(); in HandleBatchGroupCount()
269 if (kernel_input_feature_dimension > kernel_output_feature_dimension) { in HandleBatchGroupCount()
271 kernel_input_feature_dimension + 1); in HandleBatchGroupCount()
434 int64 kernel_input_feature_dim = dim_numbers.kernel_input_feature_dimension(); in HandleConvolution()
619 if (dim_numbers.kernel_input_feature_dimension() > in HandleConvolution()
622 dim_numbers.kernel_input_feature_dimension() + 1); in HandleConvolution()
Dtranspose_folding_test.cc268 EXPECT_EQ(dnums.kernel_input_feature_dimension(), in TEST_F()
273 .kernel_input_feature_dimension()); in TEST_F()
325 EXPECT_EQ(dnums.kernel_input_feature_dimension(), in TEST_F()
330 .kernel_input_feature_dimension()); in TEST_F()
Dconvolution_4d_expander.cc147 new_dim_nums.kernel_input_feature_dimension())); in ExpandInstruction()
Dtranspose_folding.cc194 transpose_dimensions[dnums.kernel_input_feature_dimension()]); in FoldTransposeIntoConvolution()
Ddot_as_convolution_util.cc64 conv_dims.kernel_input_feature_dimension(), in ParseConvolutionDimsInfo()
Dspace_to_batch_converter.cc824 .kernel_input_feature_dimension()); in CanPropagate()
2168 original_conv_dims.kernel_input_feature_dimension(); in PropagateOnBackpropFilterConv()
2225 original_conv_dims.kernel_input_feature_dimension(); in PropagateOnBackpropFilterConv()
2313 permute_dims_kernel, original_conv_dims.kernel_input_feature_dimension()); in PropagateOnBackpropFilterConv()
2392 original_conv_dims.kernel_input_feature_dimension(), in PropagateOnBackpropFilterConv()
Ddynamic_dimension_inference.cc278 hlo->mutable_operand(1), {}, dnums.kernel_input_feature_dimension()); in HandleCustomCall()
284 dnums.kernel_input_feature_dimension())) in HandleCustomCall()
580 if (dimension == dimension_numbers.kernel_input_feature_dimension()) { in HandleConvolution()
/external/tensorflow/tensorflow/compiler/mlir/hlo/tests/
Dlhlo_gpu_ops.mlir40 kernel_input_feature_dimension = 0 : i64,
75 kernel_input_feature_dimension = 2 : i64,
105 kernel_input_feature_dimension = 1 : i64,
137 kernel_input_feature_dimension = 2 : i64,
168 kernel_input_feature_dimension = 2 : i64,
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_pad_for_convolutions.cc232 new_filter_shape->set_dimensions(dnums.kernel_input_feature_dimension(), 4); in TryResolvePaddedShapesForTensorCore()
238 pad_dim(new_filter_shape, dnums.kernel_input_feature_dimension()); in TryResolvePaddedShapesForTensorCore()
320 dnums.kernel_input_feature_dimension()); // Kernel for the in TryResolvePaddedShapesForIntegerConvolution()
334 dnums.kernel_input_feature_dimension()); // Kernel for the in TryResolvePaddedShapesForIntegerConvolution()
Dgpu_conv_rewriter.cc184 auto kernel_input_feature_dim = conv_dnums.kernel_input_feature_dimension(); in MatchBackwardFilter()
508 conv_dnums.kernel_input_feature_dimension()); in MatchBackwardInput()
554 int64 input_feature_dimension = dnums.kernel_input_feature_dimension(); in MatchBackwardInput()
Dstream_executor_util.cc78 filter_layout.push_back(dnums.kernel_input_feature_dimension()); in StreamExecutorConvLayoutsToXlaLayouts()
88 filter_layout.push_back(dnums.kernel_input_feature_dimension()); in StreamExecutorConvLayoutsToXlaLayouts()
Dgpu_conv_runner.cc391 filter_shape.dimensions(dnums.kernel_input_feature_dimension())) in GetGpuConvConfig()
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/translate/
Dlayouts_and_names.mlir14 kernel_input_feature_dimension = 3 : i64,
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emission_utils.cc109 dnums.kernel_input_feature_dimension() == in PotentiallyImplementedAsEigenConvolution()
Dconv_canonicalization.cc43 auto kernel_input_feature_dim = dnums.kernel_input_feature_dimension(); in Run()
/external/tensorflow/tensorflow/compiler/xla/python/
Dxla_client.py542 self.kernel_input_feature_dimension = 0
588 dimension_numbers.kernel_input_feature_dimension = 1
601 dimension_numbers.kernel_input_feature_dimension = rhs_spec.index('I')
/external/tensorflow/tensorflow/compiler/xla/tests/
Dconvolution_dimension_numbers_test.cc116 dim_nums.kernel_input_feature_dimension(); in XLA_TEST_F()
/external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/IR/
Dhlo_ops_base_structs.td45 StructFieldAttr<"kernel_input_feature_dimension", I64Attr>,
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dconvolution_handler.cc70 rhs_to_lhs_indices[dnums.kernel_input_feature_dimension()] = in PartitionConvolutionWithBatchGroupCount()
161 rhs_to_lhs_indices[dnums.kernel_input_feature_dimension()] = in PartitionConvolutionWithFeatureGroupCount()
243 rhs_to_lhs_indices[dnums.kernel_input_feature_dimension()] = in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
533 rhs_to_lhs_indices[dnums.kernel_input_feature_dimension()] = in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS()
940 conv_dnums.kernel_input_feature_dimension()); in CreateShardedConvConvolution()
Ddot_handler.cc256 int64 kernel_input_feature_dimension = dnums.kernel_input_feature_dimension(); in GenNewConvDNums() local
263 if (rhs_concat_dim <= kernel_input_feature_dimension) { in GenNewConvDNums()
264 kernel_input_feature_dimension++; in GenNewConvDNums()
312 new_dnums.set_kernel_input_feature_dimension(kernel_input_feature_dimension); in GenNewConvDNums()
2266 dnums.kernel_input_feature_dimension(); in ConvertDimsMappingWithFeatureGroupCount()
2518 rhs.hlo()->shape().dimensions(dnums.kernel_input_feature_dimension()) > in PartitionDot()
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dattribute_exporter.cc40 input.kernel_input_feature_dimension().getValue().getSExtValue()); in ConvertConvDimensionNumbers()
Dattribute_importer.cc119 builder->getI64IntegerAttr(dnums.kernel_input_feature_dimension()), in ConvertConvDimensionNumbers()
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/hlo_to_lhlo_with_xla/
Dhlo_text_to_lhlo_no_opt.hlotxt287 // CHECK-SAME: kernel_input_feature_dimension = 1 : i64,
324 // CHECK-SAME: kernel_input_feature_dimension = 2 : i64
363 // CHECK-SAME: kernel_input_feature_dimension = 2 : i64
/external/tensorflow/tensorflow/compiler/mlir/xla/experimental/conv_emitter/
Dconv_emitter.cc542 dim_nums.kernel_input_feature_dimension(), in EmitConvolutionForwardAsMlir()
/external/tensorflow/tensorflow/compiler/tests/
Dxla_ops_test.py143 dnums.kernel_input_feature_dimension = 1

12