| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | attribute_importer.cc | 49 const xla::GatherDimensionNumbers& dnums, mlir::Builder* builder) { in ConvertGatherDimensionNumbers()
64 const xla::ScatterDimensionNumbers& dnums, mlir::Builder* builder) { in ConvertScatterDimensionNumbers()
81 const DotDimensionNumbers& dnums, mlir::Builder* builder) { in ConvertDotDimensionNumbers()
105 const xla::ConvolutionDimensionNumbers& dnums, mlir::Builder* builder) { in ConvertConvDimensionNumbers()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_sharding_util.cc | 505 const GatherDimensionNumbers& dnums = hlo->gather_dimension_numbers(); in GatherOutputSharding() local
544 const GatherDimensionNumbers& dnums = hlo->gather_dimension_numbers(); in GatherIndexSharding() local
593 const GatherDimensionNumbers& dnums = hlo.gather_dimension_numbers(); in GatherEffectiveOutputSharding() local
645 const ScatterDimensionNumbers& dnums = hlo->scatter_dimension_numbers(); in ScatterIndexSharding() local
678 const ScatterDimensionNumbers& dnums = hlo->scatter_dimension_numbers(); in ScatterDataSharding() local
713 const ScatterDimensionNumbers& dnums = hlo.scatter_dimension_numbers(); in ScatterEffectiveIndexSharding() local
756 const ScatterDimensionNumbers& dnums = hlo.scatter_dimension_numbers(); in ScatterEffectiveDataSharding() local
949 const auto& dnums = hlo.gather_dimension_numbers(); in GatherOutputShardingFromDataOperand() local
963 const auto& dnums = hlo.gather_dimension_numbers(); in GatherDataOperandShardingFromOutput() local
1009 const auto& dnums = hlo.scatter_dimension_numbers(); in ScatterOutputShardingFromUpdate() local
[all …]
|
| D | transpose_folding_test.cc | 231 auto dnums = XlaBuilder::CreateDefaultConvDimensionNumbers(); in TEST_F() local
288 auto dnums = XlaBuilder::CreateDefaultConvDimensionNumbers(); in TEST_F() local
351 auto dnums = XlaBuilder::CreateDefaultConvDimensionNumbers(); in TEST_F() local
419 auto dnums = XlaBuilder::CreateDefaultConvDimensionNumbers(); in TEST_F() local
|
| D | sharding_propagation.cc | 132 const auto& dnums = instruction->convolution_dimension_numbers(); in IsConvolutionKernelSmall() local
338 const dot_as_convolution_util::DotConvolutionDimsInfo& dnums, in InferDotShardingFromOperands()
414 auto dnums = instruction->gather_dimension_numbers(); in InferGatherParallelShardingFromOperands() local
517 const auto& dnums = instruction->convolution_dimension_numbers(); in InferConvolutionShardingFromOperands() local
857 const auto& dnums = in InferShardingFromOperands() local
1047 const dot_as_convolution_util::DotConvolutionDimsInfo& dnums, in InferDotOperandSharding()
1305 auto dnums = dot_as_convolution_util::ParseDotGeneralFromDot(&user); in GetShardingFromUser() local
|
| D | dot_as_convolution_util.cc | 157 dot_as_convolution_util::DotConvolutionDimsInfo dnums; in ParseDotGeneralFromDot() local
|
| D | dot_decomposer.cc | 180 const DotDimensionNumbers& dnums = instruction->dot_dimension_numbers(); in Run() local
|
| D | transpose_folding.cc | 156 const ConvolutionDimensionNumbers& dnums = in FoldTransposeIntoConvolution() local
|
| D | algebraic_simplifier.cc | 1762 const DotDimensionNumbers& dnums = dot->dot_dimension_numbers(); in RemoveDegenerateDimensionFromDot() local
1817 const DotDimensionNumbers& dnums = dot->dot_dimension_numbers(); in OptimizeDotOfConcat() local
1960 const DotDimensionNumbers& dnums = dot->dot_dimension_numbers(); in OptimizeDotOfGather() local
2107 auto dnums = dot->dot_dimension_numbers(); in OptimizeDotOfReorderContractingDims() local
4720 const auto& dnums = dot->dot_dimension_numbers(); in HandleReduce() local
5091 const auto& dnums = operand->dot_dimension_numbers(); in HandleTranspose() local
5147 const ConvolutionDimensionNumbers& dnums = in FoldConvInputPad() local
5213 const ConvolutionDimensionNumbers& dnums = in FoldConvFilterPad() local
5285 const auto& dnums = convolution->convolution_dimension_numbers(); in SwapConvOperands() local
5404 const ConvolutionDimensionNumbers& dnums = in SimplifyConvToDot() local
|
| D | hlo_cost_analysis_test.cc | 171 DotDimensionNumbers dnums; in TEST_F() local
207 DotDimensionNumbers dnums; in TEST_F() local
241 DotDimensionNumbers dnums; in TEST_F() local
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | convolution_test.cc | 254 ConvolutionDimensionNumbers dnums; in XLA_TEST_F() local
320 ConvolutionDimensionNumbers dnums; in RunTest() local
379 ConvolutionDimensionNumbers dnums; in RunTest() local
445 ConvolutionDimensionNumbers dnums; in RunTest() local
515 ConvolutionDimensionNumbers dnums; in RunTest() local
579 ConvolutionDimensionNumbers dnums; in RunTest() local
648 ConvolutionDimensionNumbers dnums; in RunTest() local
717 ConvolutionDimensionNumbers dnums; in RunTest() local
788 ConvolutionDimensionNumbers dnums; in RunTest() local
863 ConvolutionDimensionNumbers dnums; in RunTest() local
[all …]
|
| D | convolution_test_1d.cc | 93 ConvolutionDimensionNumbers dnums; in TestImpl() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | einsum.cc | 139 EinsumDimensionNumbers dnums; in GetEinsumDimensionNumbers() local
187 const Location& loc, EinsumDimensionNumbers& dnums, Value* lhs, Value* rhs, in transposeForBatchMatmul()
238 EinsumDimensionNumbers& dnums, Value* lhs, in reshapeForBatchMatmul()
307 EinsumDimensionNumbers dnums, in rewriteToBatchMatmul()
354 const auto& dnums = dnums_or.getValue(); in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | gather_scatter_handler.cc | 139 GatherDimensionNumbers dnums = gather->gather_dimension_numbers(); in PartitionIndexOnlyPartition() local
185 GatherDimensionNumbers dnums = gather->gather_dimension_numbers(); in ParititonPassthroughOperand() local
220 GatherDimensionNumbers dnums = gather->gather_dimension_numbers(); in ParititonTrivialIndexedOperandDimension() local
327 GatherDimensionNumbers dnums = gather->gather_dimension_numbers(); in PartitionIndexParallelDimensions() local
509 auto dnums = scatter->scatter_dimension_numbers(); in HandleScatter() local
673 const auto& dnums = gather->gather_dimension_numbers(); in HandleGather() local
|
| D | convolution_handler.cc | 55 const auto& dnums = original_hlo->convolution_dimension_numbers(); in PartitionConvolutionWithBatchGroupCount() local
147 const auto& dnums = original_hlo->convolution_dimension_numbers(); in PartitionConvolutionWithFeatureGroupCount() local
239 const auto& dnums = original_hlo->convolution_dimension_numbers(); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS() local
526 const auto& dnums = original_hlo->convolution_dimension_numbers(); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS() local
748 const auto& dnums = original_hlo->convolution_dimension_numbers(); in PartitionConvolutionTiledOutput() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | conv_canonicalization_test.cc | 70 ConvolutionDimensionNumbers dnums; in TEST_F() local
133 ConvolutionDimensionNumbers dnums; in TEST_F() local
|
| D | ir_emission_utils.cc | 83 const ConvolutionDimensionNumbers& dnums = in PotentiallyImplementedAsEigenConvolution() local
|
| D | conv_canonicalization.cc | 39 const ConvolutionDimensionNumbers& dnums = in Run() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_conv_rewriter.cc | 43 const ConvolutionDimensionNumbers& dnums, in CreateGpuConv()
122 const ConvolutionDimensionNumbers& dnums = in CanImplementAsGpuForwardConv() local
344 ConvolutionDimensionNumbers dnums = conv->convolution_dimension_numbers(); in MatchBackwardInput() local
619 ConvolutionDimensionNumbers dnums; in CreateCustomCallHelper() local
|
| D | stream_executor_util.cc | 49 StreamExecutorConvLayoutsToXlaLayouts(const ConvolutionDimensionNumbers& dnums, in StreamExecutorConvLayoutsToXlaLayouts()
124 XlaConvLayoutsToStreamExecutorLayouts(const ConvolutionDimensionNumbers& dnums, in XlaConvLayoutsToStreamExecutorLayouts()
|
| D | cudnn_pad_for_convolutions.cc | 182 const auto& dnums = conv->convolution_dimension_numbers(); in TryResolvePaddedShapesForTensorCore() local
299 const auto& dnums = conv->convolution_dimension_numbers(); in TryResolvePaddedShapesForIntegerConvolution() local
|
| D | gpu_conv_runner.h | 133 ConvolutionDimensionNumbers dnums; member
|
| D | gpu_conv_runner.cc | 314 const ConvolutionDimensionNumbers& dnums = desc.dnums; in GetGpuConvConfig() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | reverse_sequence_op.cc | 100 xla::GatherDimensionNumbers dnums; in Compile() local
|
| D | conv_op_helpers.cc | 327 xla::ConvolutionDimensionNumbers dnums; in MakeXlaBackpropInputConvOp() local
420 xla::ConvolutionDimensionNumbers dnums; in MakeXlaBackpropFilterConvOp() local
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder_test.cc | 783 ConvolutionDimensionNumbers dnums; in TEST_F() local
831 DotDimensionNumbers dnums; in TEST_F() local
1090 DotDimensionNumbers dnums; in TEST_F() local
1109 ConvolutionDimensionNumbers dnums; in TEST_F() local
|