Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/compiler/xla/service/
Dconvolution_group_converter.cc221 int64 output_feature_dimension = dim_numbers.output_feature_dimension(); in HandleBatchGroupCount() local
277 if (d > output_feature_dimension) { in HandleBatchGroupCount()
281 dim_numbers.add_output_spatial_dimensions(output_feature_dimension); in HandleBatchGroupCount()
282 dim_numbers.set_output_feature_dimension(output_feature_dimension + 1); in HandleBatchGroupCount()
283 if (output_batch_dimension > output_feature_dimension) { in HandleBatchGroupCount()
327 output_feature_dimension, batch_group_count, add); in HandleBatchGroupCount()
545 if (i == dim_numbers.output_feature_dimension()) { in HandleConvolution()
553 dim_numbers.output_feature_dimension()) { in HandleConvolution()
558 if (dim > dim_numbers.output_feature_dimension()) { in HandleConvolution()
563 dim_numbers.output_feature_dimension() + 1); in HandleConvolution()
[all …]
Dconvolution_4d_expander.cc139 removed_output_dimensions, new_dim_nums.output_feature_dimension())); in ExpandInstruction()
Ddot_as_convolution_util.cc62 conv_dims.output_feature_dimension(), -1}); in ParseConvolutionDimsInfo()
Dhlo_cost_analysis.cc581 const int64 output_feature_dim = dnums.output_feature_dimension(); in HandleConvolution()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_conv_rewriter.cc188 auto output_feature_dim = conv_dnums.output_feature_dimension(); in MatchBackwardFilter()
515 dnums.set_input_feature_dimension(conv_dnums.output_feature_dimension()); in MatchBackwardInput()
555 int64 output_feature_dimension = dnums.kernel_output_feature_dimension(); in MatchBackwardInput() local
558 if (std::abs(input_feature_dimension - output_feature_dimension) != 1) { in MatchBackwardInput()
563 int64 output_features = rhs->shape().dimensions(output_feature_dimension); in MatchBackwardInput()
586 transpose_dims.insert(transpose_dims.begin() + output_feature_dimension, in MatchBackwardInput()
593 transpose_reshape_dims.begin() + output_feature_dimension, num_groups); in MatchBackwardInput()
601 new_shape.DeleteDimension(output_feature_dimension); in MatchBackwardInput()
602 new_shape.set_dimensions(output_feature_dimension, in MatchBackwardInput()
Dcudnn_pad_for_convolutions.cc229 new_output_shape->dimensions(dnums.output_feature_dimension()); in TryResolvePaddedShapesForTensorCore()
240 pad_dim(new_output_shape, dnums.output_feature_dimension()); in TryResolvePaddedShapesForTensorCore()
327 dnums.output_feature_dimension()); // Output feature maps in TryResolvePaddedShapesForIntegerConvolution()
343 dnums.output_feature_dimension()); // Optional side input in TryResolvePaddedShapesForIntegerConvolution()
346 dnums.output_feature_dimension()); // Output feature maps in TryResolvePaddedShapesForIntegerConvolution()
Dstream_executor_util.cc100 output_layout.push_back(dnums.output_feature_dimension()); in StreamExecutorConvLayoutsToXlaLayouts()
110 output_layout.push_back(dnums.output_feature_dimension()); in StreamExecutorConvLayoutsToXlaLayouts()
Dcudnn_fused_conv_rewriter.cc171 conv->convolution_dimension_numbers().output_feature_dimension()) { in FindConvWithRelu()
225 conv->convolution_dimension_numbers().output_feature_dimension()); in TryRewriteToCudnnForwardRelu()
Dgpu_conv_runner.cc420 output_shape.dimensions(dnums.output_feature_dimension())) in GetGpuConvConfig()
/external/tensorflow/tensorflow/compiler/mlir/hlo/tests/
Dlhlo_gpu_ops.mlir44 output_feature_dimension = 1 : i64,
79 output_feature_dimension = 3 : i64,
109 output_feature_dimension = 1 : i64,
141 output_feature_dimension = 1 : i64,
172 output_feature_dimension = 1 : i64,
Dlhlo-legalize-to-linalg.mlir908 …ense<[0, 1]> : tensor<2xi64>, output_batch_dimension = 0 : i64, output_feature_dimension = 3 : i64…
915 …ense<[0, 1]> : tensor<2xi64>, output_batch_dimension = 0 : i64, output_feature_dimension = 3 : i64…
Dhlo-legalize-to-lhlo.mlir552 output_feature_dimension = 3 : i64,
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/translate/
Dlayouts_and_names.mlir18 output_feature_dimension = 1 : i64,
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emission_utils.cc107 dnums.output_feature_dimension() == in PotentiallyImplementedAsEigenConvolution()
Dconv_canonicalization.cc104 auto output_feature_dim = dnums.output_feature_dimension(); in Run()
/external/tensorflow/tensorflow/compiler/xla/python/
Dxla_client.py546 self.output_feature_dimension = 0
586 dimension_numbers.output_feature_dimension = 1
599 dimension_numbers.output_feature_dimension = out_spec.index('C')
/external/tensorflow/tensorflow/compiler/xla/tests/
Dconvolution_dimension_numbers_test.cc111 dim_nums.set_output_batch_dimension(dim_nums.output_feature_dimension()); in XLA_TEST_F()
/external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/IR/
Dhlo_ops_base_structs.td49 StructFieldAttr<"output_feature_dimension", I64Attr>,
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dattribute_exporter.cc51 input.output_feature_dimension().getValue().getSExtValue()); in ConvertConvDimensionNumbers()
Dattribute_importer.cc123 builder->getI64IntegerAttr(dnums.output_feature_dimension()), in ConvertConvDimensionNumbers()
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dconvolution_handler.cc86 dnums.output_feature_dimension(); in PartitionConvolutionWithBatchGroupCount()
177 dnums.output_feature_dimension(); in PartitionConvolutionWithFeatureGroupCount()
751 if (output_sharding.tile_assignment().dim(dnums.output_feature_dimension()) > in PartitionConvolutionTiledOutput()
761 dnums.output_feature_dimension(); in PartitionConvolutionTiledOutput()
Ddot_handler.cc282 int64 output_feature_dimension = dnums.output_feature_dimension(); in GenNewConvDNums() local
293 if (output_slice_dim <= output_feature_dimension) { in GenNewConvDNums()
294 output_feature_dimension++; in GenNewConvDNums()
319 new_dnums.set_output_feature_dimension(output_feature_dimension); in GenNewConvDNums()
2258 new_dims_mapping.batch_dims.back().output = dnums.output_feature_dimension(); in ConvertDimsMappingWithFeatureGroupCount()
2282 new_dims_mapping.batch_dims.back().output = dnums.output_feature_dimension(); in ConvertDimsMappingWithBatchGroupCount()
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/hlo_to_lhlo_with_xla/
Dhlo_text_to_lhlo_no_opt.hlotxt291 // CHECK-SAME: output_feature_dimension = 1 : i64
328 // CHECK-SAME: output_feature_dimension = 1 : i64
367 // CHECK-SAME: output_feature_dimension = 1 : i64
/external/tensorflow/tensorflow/compiler/mlir/xla/experimental/conv_emitter/
Dconv_emitter.cc547 dim_nums.output_feature_dimension(), dim_nums.output_spatial_dimensions(), in EmitConvolutionForwardAsMlir()
/external/tensorflow/tensorflow/compiler/tests/
Dxla_ops_test.py141 dnums.output_feature_dimension = 1

12