Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dstream_executor_util.cc38 StreamExecutorConvLayoutsToXlaLayouts(const ConvolutionDimensionNumbers& dnums, in StreamExecutorConvLayoutsToXlaLayouts()
113 XlaConvLayoutsToStreamExecutorLayouts(const ConvolutionDimensionNumbers& dnums, in XlaConvLayoutsToStreamExecutorLayouts()
Dcudnn_conv_rewriter.cc42 const ConvolutionDimensionNumbers& dnums, in CreateCudnnConv()
68 const ConvolutionDimensionNumbers& dnums = in CanImplementAsCudnnForwardConv() local
273 ConvolutionDimensionNumbers dnums = conv->convolution_dimension_numbers(); in MatchBackwardInput() local
504 ConvolutionDimensionNumbers dnums; in RunOnInstruction() local
Dcudnn_conv_runner.cc70 const ConvolutionDimensionNumbers* dnums; member
125 const ConvolutionDimensionNumbers& dnums = *params.dnums; in RunCudnnConvImpl() local
Dcudnn_conv_pad_for_tensor_cores.cc123 const auto& dnums = conv->convolution_dimension_numbers(); in PadForTensorCores() local
/external/tensorflow/tensorflow/compiler/xla/tests/
Dconvolution_test.cc422 ConvolutionDimensionNumbers dnums; in XLA_TEST_F() local
488 ConvolutionDimensionNumbers dnums; in RunTest() local
547 ConvolutionDimensionNumbers dnums; in RunTest() local
613 ConvolutionDimensionNumbers dnums; in RunTest() local
683 ConvolutionDimensionNumbers dnums; in RunTest() local
747 ConvolutionDimensionNumbers dnums; in RunTest() local
816 ConvolutionDimensionNumbers dnums; in RunTest() local
885 ConvolutionDimensionNumbers dnums; in RunTest() local
956 ConvolutionDimensionNumbers dnums; in RunTest() local
1031 ConvolutionDimensionNumbers dnums; in RunTest() local
[all …]
Ddot_operation_test.cc614 DotDimensionNumbers dnums; in XLA_TYPED_TEST() local
651 DotDimensionNumbers dnums; in XLA_TYPED_TEST() local
685 DotDimensionNumbers dnums; in XLA_TYPED_TEST() local
720 DotDimensionNumbers dnums; in XLA_TYPED_TEST() local
Dconvolution_variants_test.cc980 ConvolutionDimensionNumbers dnums; in XLA_TEST_F() local
1024 ConvolutionDimensionNumbers dnums; in XLA_TEST_F() local
1068 ConvolutionDimensionNumbers dnums; in XLA_TEST_F() local
1109 ConvolutionDimensionNumbers dnums; in XLA_TEST_F() local
/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.cc83 const ConvolutionDimensionNumbers& dnums = in PotentiallyImplementedAsEigenConvolution() local
Dconv_canonicalization.cc38 const ConvolutionDimensionNumbers& dnums = in Run() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dtranspose_folding_test.cc230 auto dnums = XlaBuilder::CreateDefaultConvDimensionNumbers(); in TEST_F() local
286 auto dnums = XlaBuilder::CreateDefaultConvDimensionNumbers(); in TEST_F() local
348 auto dnums = XlaBuilder::CreateDefaultConvDimensionNumbers(); in TEST_F() local
415 auto dnums = XlaBuilder::CreateDefaultConvDimensionNumbers(); in TEST_F() local
Ddot_decomposer.cc36 const DotDimensionNumbers& dnums = dot->dot_dimension_numbers(); in DecomposeBatchDot() local
297 const DotDimensionNumbers& dnums = instruction->dot_dimension_numbers(); in Run() local
334 const DotDimensionNumbers& dnums = instruction->dot_dimension_numbers(); in Run() local
Ddynamic_padder_test.cc118 auto dnums = XlaBuilder::CreateDefaultConvDimensionNumbers(0); in TEST_F() local
161 auto dnums = XlaBuilder::CreateDefaultConvDimensionNumbers(0); in TEST_F() local
Dtranspose_folding.cc127 const ConvolutionDimensionNumbers& dnums = in FoldTransposeIntoConvolution() local
Dhlo_cost_analysis_test.cc167 DotDimensionNumbers dnums; in TEST_F() local
196 DotDimensionNumbers dnums; in TEST_F() local
223 DotDimensionNumbers dnums; in TEST_F() local
Dhlo_instructions.h1042 const ConvolutionDimensionNumbers& dnums) { in set_convolution_dimension_numbers()
1197 const ConvolutionDimensionNumbers& dnums) { in set_convolution_dimension_numbers()
Dalgebraic_simplifier.cc1146 const auto& dnums = dot->dot_dimension_numbers(); in HandleDotStrengthReduction() local
1365 const DotDimensionNumbers& dnums = dot->dot_dimension_numbers(); in OptimizeDotOfConcat() local
1506 const DotDimensionNumbers& dnums = dot->dot_dimension_numbers(); in OptimizeDotOfGather() local
3398 const ConvolutionDimensionNumbers& dnums = in FoldConvInputPad() local
3464 const ConvolutionDimensionNumbers& dnums = in FoldConvFilterPad() local
3531 const ConvolutionDimensionNumbers& dnums = in SimplifyConvToDot() local
Dhlo_evaluator_test.cc914 ConvolutionDimensionNumbers dnums; in TEST_P() local
978 ConvolutionDimensionNumbers dnums = in TEST_P() local
1050 ConvolutionDimensionNumbers dnums; in TEST_P() local
1128 ConvolutionDimensionNumbers dnums; in TEST_P() local
1200 ConvolutionDimensionNumbers dnums = in TEST_P() local
1264 ConvolutionDimensionNumbers dnums = in TEST_P() local
1336 ConvolutionDimensionNumbers dnums = in TEST_P() local
1371 ConvolutionDimensionNumbers dnums; in TEST_P() local
Dhlo_cost_analysis.cc242 const DotDimensionNumbers& dnums = dot->dot_dimension_numbers(); in HandleDot() local
442 const auto& dnums = convolution->convolution_dimension_numbers(); in HandleConvolution() local
Dtriangular_solve_expander.cc221 DotDimensionNumbers dnums; in InvertDiagonalBlocks() local
Dhlo_evaluator_typed_visitor.h1024 const auto& dnums = conv->convolution_dimension_numbers(); in HandleConvolution() local
1221 const auto& dnums = dot->dot_dimension_numbers(); in HandleDot() local
1282 const auto& dnums = dot->dot_dimension_numbers(); in HandleDotSlowPath() local
Dhlo_parser.cc1066 optional<ConvolutionDimensionNumbers> dnums; in ParseInstructionRhs() local
1487 optional<ConvolutionDimensionNumbers> dnums; in ParseInstructionRhs() local
3093 ConvolutionDimensionNumbers* dnums) { in ParseConvolutionDimensionNumbers()
4025 ConvolutionDimensionNumbers dnums; in ParseConvolutionDimensionNumbersOnly() local
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dreverse_sequence_op.cc100 xla::GatherDimensionNumbers dnums; in Compile() local
Dconv_op_helpers.cc351 xla::ConvolutionDimensionNumbers dnums; in MakeXlaBackpropInputConvOp() local
423 xla::ConvolutionDimensionNumbers dnums; in MakeXlaBackpropFilterConvOp() local
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dmatrix.cc195 DotDimensionNumbers dnums; in Einsum() local

12