Home
last modified time | relevance | path

Searched defs:dnums (Results 1 – 25 of 62) sorted by relevance

123

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_vectorize_convolutions_test.cc84 const ConvolutionDimensionNumbers& dnums = in TEST_F() local
157 const ConvolutionDimensionNumbers& dnums = in TEST_F() local
210 const ConvolutionDimensionNumbers& dnums = in TEST_F() local
263 const ConvolutionDimensionNumbers& dnums = in TEST_F() local
493 const ConvolutionDimensionNumbers& dnums = in TEST_F() local
561 const ConvolutionDimensionNumbers& dnums = in TEST_F() local
629 const ConvolutionDimensionNumbers& dnums = in TEST_F() local
Dcudnn_simplify_padding.cc58 const ConvolutionDimensionNumbers& dnums = in FindOutputVectCDim() local
72 const ConvolutionDimensionNumbers& dnums = in FindKernelVectCDim() local
90 const ConvolutionDimensionNumbers& dnums = in NumTrailingZeroOutputFeatures() local
272 const auto& dnums = conv->convolution_dimension_numbers(); in TrySimplifyPadding() local
Dgpu_conv_rewriter.cc43 const ConvolutionDimensionNumbers& dnums = in CanImplementAsGpuForwardConv() local
276 ConvolutionDimensionNumbers dnums = conv->convolution_dimension_numbers(); in MatchBackwardInput() local
563 const ConvolutionDimensionNumbers& dnums, in CreateGpuConv()
668 ConvolutionDimensionNumbers dnums; in CreateCustomCallHelper() local
Dcudnn_vectorize_convolutions.cc217 ConvolutionDimensionNumbers dnums) { in VectorizeDnums()
266 const auto& dnums = conv->convolution_dimension_numbers(); in TryRevectorizeConv() local
403 const auto& dnums = conv->convolution_dimension_numbers(); in TryVectorizeConv() local
Dstream_executor_util.cc67 StreamExecutorConvLayoutsToXlaLayouts(const ConvolutionDimensionNumbers& dnums, in StreamExecutorConvLayoutsToXlaLayouts()
169 XlaConvShapesToStreamExecutorLayouts(const ConvolutionDimensionNumbers& dnums, in XlaConvShapesToStreamExecutorLayouts()
295 FindVectorizedFeatureDims(const ConvolutionDimensionNumbers& dnums, in FindVectorizedFeatureDims()
Dcudnn_support_utils.cc34 const auto& dnums = conv.convolution_dimension_numbers(); in CudnnSupportsOptimizedIntegerConvolution() local
Dcudnn_pad_for_convolutions.cc184 const auto& dnums = conv->convolution_dimension_numbers(); in TryResolvePaddedShapesForTensorCore() local
314 const auto& dnums = conv->convolution_dimension_numbers(); in TryResolvePaddedShapesForIntegerConvolution() local
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dattribute_importer.cc47 const xla::GatherDimensionNumbers& dnums, mlir::Builder* builder) { in ConvertGatherDimensionNumbers()
60 const xla::ScatterDimensionNumbers& dnums, mlir::Builder* builder) { in ConvertScatterDimensionNumbers()
74 const DotDimensionNumbers& dnums, mlir::Builder* builder) { in ConvertDotDimensionNumbers()
86 const xla::ConvolutionDimensionNumbers& dnums, mlir::Builder* builder) { in ConvertConvDimensionNumbers()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dconvolution_test.cc269 ConvolutionDimensionNumbers dnums; in XLA_TEST_F() local
334 ConvolutionDimensionNumbers dnums; in RunTest() local
391 ConvolutionDimensionNumbers dnums; in RunTest() local
455 ConvolutionDimensionNumbers dnums; in RunTest() local
523 ConvolutionDimensionNumbers dnums; in RunTest() local
585 ConvolutionDimensionNumbers dnums; in RunTest() local
652 ConvolutionDimensionNumbers dnums; in RunTest() local
718 ConvolutionDimensionNumbers dnums; in RunTest() local
786 ConvolutionDimensionNumbers dnums; in RunTest() local
859 ConvolutionDimensionNumbers dnums; in RunTest() local
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_sharding_util.cc590 const GatherDimensionNumbers& dnums = hlo->gather_dimension_numbers(); in GatherOutputSharding() local
634 const GatherDimensionNumbers& dnums = hlo->gather_dimension_numbers(); in GatherIndexSharding() local
684 const GatherDimensionNumbers& dnums = hlo.gather_dimension_numbers(); in GatherEffectiveOutputSharding() local
736 const ScatterDimensionNumbers& dnums = scatter->scatter_dimension_numbers(); in ScatterIndexSharding() local
779 const ScatterDimensionNumbers& dnums = scatter->scatter_dimension_numbers(); in ScatterDataSharding() local
822 const ScatterDimensionNumbers& dnums = scatter.scatter_dimension_numbers(); in ScatterEffectiveIndexSharding() local
865 const ScatterDimensionNumbers& dnums = scatter.scatter_dimension_numbers(); in ScatterEffectiveDataSharding() local
1065 const auto& dnums = hlo.gather_dimension_numbers(); in GatherOutputShardingFromDataOperand() local
1079 const auto& dnums = hlo.gather_dimension_numbers(); in GatherDataOperandShardingFromOutput() local
1120 const ScatterDimensionNumbers& dnums) { in GetScatterSliceSize()
[all …]
Dtranspose_folding_test.cc237 auto dnums = XlaBuilder::CreateDefaultConvDimensionNumbers(); in TEST_F() local
295 auto dnums = XlaBuilder::CreateDefaultConvDimensionNumbers(); in TEST_F() local
359 auto dnums = XlaBuilder::CreateDefaultConvDimensionNumbers(); in TEST_F() local
428 auto dnums = XlaBuilder::CreateDefaultConvDimensionNumbers(); in TEST_F() local
Ddot_merger.cc55 const DotDimensionNumbers& dnums = dot->dot_dimension_numbers(); in IsCanonicalDot() local
149 const DotDimensionNumbers& dnums = a->dot_dimension_numbers(); in TryMergeSameOperand() local
Dsharding_propagation.cc133 const auto& dnums = instruction->convolution_dimension_numbers(); in IsConvolutionKernelSmall() local
388 const dot_as_convolution_util::DotConvolutionDimsInfo& dnums, in InferDotShardingFromOperands()
464 auto dnums = instruction->gather_dimension_numbers(); in InferGatherParallelShardingFromOperands() local
572 const auto& dnums = instruction->convolution_dimension_numbers(); in InferConvolutionShardingFromOperands() local
639 const dot_as_convolution_util::DotConvolutionDimsInfo& dnums, in InferDotOperandSharding()
1509 auto dnums = dot_as_convolution_util::ParseDotGeneralFromDot(&user); in GetShardingFromUser() local
2077 const auto& dnums = in InferShardingFromOperands() local
Ddot_as_convolution_util.cc184 dot_as_convolution_util::DotConvolutionDimsInfo dnums; in ParseDotGeneralFromDot() local
Ddot_decomposer.cc197 const DotDimensionNumbers& dnums = instruction->dot_dimension_numbers(); in Run() local
Dtranspose_folding.cc123 const ConvolutionDimensionNumbers& dnums = in FoldTransposeIntoConvolution() local
Dalgebraic_simplifier.cc1976 const DotDimensionNumbers& dnums = dot->dot_dimension_numbers(); in RemoveDegenerateDimensionFromDot() local
2032 const auto& dnums = dot->dot_dimension_numbers(); in RemoveTransposesFromDotOperands() local
2125 const DotDimensionNumbers& dnums = dot->dot_dimension_numbers(); in OptimizeDotOfConcat() local
2267 const DotDimensionNumbers& dnums = dot->dot_dimension_numbers(); in OptimizeDotOfGather() local
2414 auto dnums = dot->dot_dimension_numbers(); in OptimizeDotOfReorderContractingDims() local
2614 const auto& dnums = dot->dot_dimension_numbers(); in HandleDot() local
5520 const auto& dnums = dot->dot_dimension_numbers(); in HandleReduce() local
6067 const auto& dnums = dot->dot_dimension_numbers(); in HandleTranspose() local
6285 const ConvolutionDimensionNumbers& dnums = in FoldConvInputPad() local
6343 const ConvolutionDimensionNumbers& dnums = in FoldConvFilterPad() local
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dconv_canonicalization_test.cc70 ConvolutionDimensionNumbers dnums; in TEST_F() local
133 ConvolutionDimensionNumbers dnums; in TEST_F() local
Dir_emission_utils.cc82 const ConvolutionDimensionNumbers& dnums = in PotentiallyImplementedAsEigenConvolution() local
Dconv_canonicalization.cc41 const ConvolutionDimensionNumbers& dnums = in Run() local
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dgather_scatter_handler.cc172 GatherDimensionNumbers dnums = gather->gather_dimension_numbers(); in PartitionGatherIndexPassthroughPartition() local
279 GatherDimensionNumbers dnums = gather->gather_dimension_numbers(); in PartitionGatherPassthroughOperand() local
380 GatherDimensionNumbers dnums = gather->gather_dimension_numbers(); in PartitionGatherTrivialIndexedOperandDimension() local
491 GatherDimensionNumbers dnums = gather->gather_dimension_numbers(); in PartitionGatherIndexParallelDimensions() local
687 const auto& dnums = gather->gather_dimension_numbers(); in HandleGather() local
782 auto dnums = scatter->scatter_dimension_numbers(); in PartitionScatterPassthroughOperand() local
818 auto dnums = scatter->scatter_dimension_numbers(); in PartitionScatterIndexPassthroughPartition() local
957 auto dnums = scatter->scatter_dimension_numbers(); in PartitionScatterTrivialIndexedOperandDimension() local
1045 auto dnums = scatter->scatter_dimension_numbers(); in HandleScatter() local
Dconvolution_handler.cc55 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
528 const auto& dnums = original_hlo->convolution_dimension_numbers(); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS() local
750 const auto& dnums = original_hlo->convolution_dimension_numbers(); in PartitionConvolutionTiledOutput() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Deinsum.cc155 EinsumDimensionNumbers& dnums, Location loc, PatternRewriter* rewriter) { in createOutputReshapeOpForDynamic()
384 EinsumDimensionNumbers dnums; in GetEinsumDimensionNumbers() local
434 const Location& loc, EinsumDimensionNumbers& dnums, Value* lhs, Value* rhs, in transposeForBatchMatmul()
501 EinsumDimensionNumbers& dnums, Value* lhs, in reshapeForBatchMatmul()
629 EinsumDimensionNumbers dnums, in rewriteToBatchMatmul()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dreverse_sequence_op.cc101 xla::GatherDimensionNumbers dnums; in Compile() local
Dconv_op_helpers.cc327 xla::ConvolutionDimensionNumbers dnums; in MakeXlaBackpropInputConvOp() local
420 xla::ConvolutionDimensionNumbers dnums; in MakeXlaBackpropFilterConvOp() local

123