Home
last modified time | relevance | path

Searched refs:kernel_output_feature_dimension (Results 1 – 25 of 40) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/xla/service/
Dconvolution_group_converter.cc225 const int64 kernel_output_feature_dimension = in HandleBatchGroupCount() local
226 dim_numbers.kernel_output_feature_dimension(); in HandleBatchGroupCount()
231 filter->shape().dimensions(kernel_output_feature_dimension); in HandleBatchGroupCount()
257 kernel_sizes[kernel_output_feature_dimension] /= batch_group_count; in HandleBatchGroupCount()
258 kernel_sizes.insert(kernel_sizes.begin() + kernel_output_feature_dimension, in HandleBatchGroupCount()
262 if (d > kernel_output_feature_dimension) { in HandleBatchGroupCount()
266 dim_numbers.add_kernel_spatial_dimensions(kernel_output_feature_dimension); in HandleBatchGroupCount()
268 kernel_output_feature_dimension + 1); in HandleBatchGroupCount()
269 if (kernel_input_feature_dimension > kernel_output_feature_dimension) { in HandleBatchGroupCount()
437 dim_numbers.kernel_output_feature_dimension(); in HandleConvolution()
[all …]
Dtranspose_folding_test.cc270 .kernel_output_feature_dimension()); in TEST_F()
271 EXPECT_EQ(dnums.kernel_output_feature_dimension(), in TEST_F()
327 .kernel_output_feature_dimension()); in TEST_F()
332 dnums.kernel_output_feature_dimension(), in TEST_F()
Dconvolution_4d_expander.cc150 new_dim_nums.kernel_output_feature_dimension())); in ExpandInstruction()
Dtranspose_folding.cc196 transpose_dimensions[dnums.kernel_output_feature_dimension()]); in FoldTransposeIntoConvolution()
Ddot_as_convolution_util.cc61 {-1, conv_dims.kernel_output_feature_dimension(), in ParseConvolutionDimsInfo()
Dspace_to_batch_converter.cc503 if (i == dim_numbers.kernel_output_feature_dimension()) { in BringSpaceNextToBatch()
2227 original_conv_dims.kernel_output_feature_dimension(); in PropagateOnBackpropFilterConv()
2317 original_conv_dims.kernel_output_feature_dimension()); in PropagateOnBackpropFilterConv()
Dshape_inference.cc1691 window_dnums[1] = dnums.kernel_output_feature_dimension(); in InferConvolveShape()
1748 rhs.dimensions(dnums.kernel_output_feature_dimension()); in InferConvolveShape()
1850 } else if (i == dnums.kernel_output_feature_dimension()) { in InferConvolveShape()
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dconvolution_handler.cc60 rhs.base_shape().dimensions(dnums.kernel_output_feature_dimension()); in PartitionConvolutionWithBatchGroupCount()
68 rhs_to_lhs_indices[dnums.kernel_output_feature_dimension()] = in PartitionConvolutionWithBatchGroupCount()
106 dnums.kernel_output_feature_dimension()) == in PartitionConvolutionWithBatchGroupCount()
151 rhs.base_shape().dimensions(dnums.kernel_output_feature_dimension()); in PartitionConvolutionWithFeatureGroupCount()
159 rhs_to_lhs_indices[dnums.kernel_output_feature_dimension()] = in PartitionConvolutionWithFeatureGroupCount()
197 dnums.kernel_output_feature_dimension()) == in PartitionConvolutionWithFeatureGroupCount()
241 rhs_to_lhs_indices[dnums.kernel_output_feature_dimension()] = in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
265 dnums.kernel_output_feature_dimension()) != 1; in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
288 dnums.kernel_output_feature_dimension()) > 1)) { in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
296 dnums.kernel_output_feature_dimension()) > 1)) { in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
[all …]
Ddot_handler.cc257 int64 kernel_output_feature_dimension = in GenNewConvDNums() local
258 dnums.kernel_output_feature_dimension(); in GenNewConvDNums()
266 if (rhs_concat_dim <= kernel_output_feature_dimension) { in GenNewConvDNums()
267 kernel_output_feature_dimension++; in GenNewConvDNums()
314 kernel_output_feature_dimension); in GenNewConvDNums()
2257 dnums.kernel_output_feature_dimension(); in ConvertDimsMappingWithFeatureGroupCount()
2281 dnums.kernel_output_feature_dimension(); in ConvertDimsMappingWithBatchGroupCount()
/external/tensorflow/tensorflow/compiler/mlir/hlo/tests/
Dlhlo_gpu_ops.mlir41 kernel_output_feature_dimension = 1 : i64,
76 kernel_output_feature_dimension = 3 : i64,
106 kernel_output_feature_dimension = 0 : i64,
138 kernel_output_feature_dimension = 3 : i64,
169 kernel_output_feature_dimension = 3 : i64,
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_pad_for_convolutions.cc239 pad_dim(new_filter_shape, dnums.kernel_output_feature_dimension()); in TryResolvePaddedShapesForTensorCore()
324 dnums.kernel_output_feature_dimension()); // Kernel for the output in TryResolvePaddedShapesForIntegerConvolution()
338 dnums.kernel_output_feature_dimension()); // Kernel for the output in TryResolvePaddedShapesForIntegerConvolution()
Dgpu_conv_rewriter.cc185 auto kernel_output_feature_dim = conv_dnums.kernel_output_feature_dimension(); in MatchBackwardFilter()
506 conv_dnums.kernel_output_feature_dimension()); in MatchBackwardInput()
555 int64 output_feature_dimension = dnums.kernel_output_feature_dimension(); in MatchBackwardInput()
Dstream_executor_util.cc77 filter_layout.push_back(dnums.kernel_output_feature_dimension()); in StreamExecutorConvLayoutsToXlaLayouts()
84 filter_layout.push_back(dnums.kernel_output_feature_dimension()); in StreamExecutorConvLayoutsToXlaLayouts()
Dgpu_conv_runner.cc393 filter_shape.dimensions(dnums.kernel_output_feature_dimension())); in GetGpuConvConfig()
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/translate/
Dlayouts_and_names.mlir15 kernel_output_feature_dimension = 0 : i64,
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emission_utils.cc111 dnums.kernel_output_feature_dimension() == in PotentiallyImplementedAsEigenConvolution()
Dconv_canonicalization.cc44 auto kernel_output_feature_dim = dnums.kernel_output_feature_dimension(); in Run()
/external/tensorflow/tensorflow/compiler/xla/python/
Dxla_client.py543 self.kernel_output_feature_dimension = 0
587 dimension_numbers.kernel_output_feature_dimension = 0
600 dimension_numbers.kernel_output_feature_dimension = rhs_spec.index('O')
/external/tensorflow/tensorflow/compiler/xla/tests/
Dconvolution_dimension_numbers_test.cc118 dim_nums.kernel_output_feature_dimension()); in XLA_TEST_F()
/external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/IR/
Dhlo_ops_base_structs.td46 StructFieldAttr<"kernel_output_feature_dimension", I64Attr>,
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dattribute_exporter.cc42 input.kernel_output_feature_dimension().getValue().getSExtValue()); in ConvertConvDimensionNumbers()
Dattribute_importer.cc120 builder->getI64IntegerAttr(dnums.kernel_output_feature_dimension()), in ConvertConvDimensionNumbers()
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/hlo_to_lhlo_with_xla/
Dhlo_text_to_lhlo_no_opt.hlotxt288 // CHECK_SAME: kernel_output_feature_dimension = 0 : i64,
325 // CHECK-SAME: kernel_output_feature_dimension = 3 : i64
364 // CHECK-SAME: kernel_output_feature_dimension = 3 : i64
/external/tensorflow/tensorflow/compiler/mlir/xla/experimental/conv_emitter/
Dconv_emitter.cc541 conv->operand(1)->shape(), dim_nums.kernel_output_feature_dimension(), in EmitConvolutionForwardAsMlir()
/external/tensorflow/tensorflow/compiler/tests/
Dxla_ops_test.py142 dnums.kernel_output_feature_dimension = 0

12