Home
last modified time | relevance | path

Searched defs:dim_nums (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dconvolution_4d_expander.cc39 const ConvolutionDimensionNumbers& dim_nums = in InstructionMatchesPattern() local
59 ConvolutionDimensionNumbers dim_nums = in ExpandInstruction() local
Dselect_and_scatter_expander.cc182 ScatterDimensionNumbers dim_nums = in ExpandInstruction() local
Dhlo_matchers.cc232 const DotDimensionNumbers& dim_nums = instruction->dot_dimension_numbers(); in MatchAndExplain() local
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Ddot_op_emitter_internal.h41 DotDimensionNumbers dim_nums; member
Ddot_op_emitter.cc72 DotDimensionNumbers dim_nums; member
618 const DotDimensionNumbers& dim_nums = dot_info_.dim_nums; in EmitNaiveLlvmIrGemm() local
1008 const DotDimensionNumbers& dim_nums = dot_info_.dim_nums; in GetMatMultDims() local
1038 const DotDimensionNumbers& dim_nums = dot_info_.dim_nums; in GetBatchMatMultDims() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgemm_broadcast_folding_rewriter.cc51 DotDimensionNumbers *dim_nums = config.mutable_dot_dimension_numbers(); in HandleCustomCall() local
/external/tensorflow/tensorflow/compiler/xla/tests/
Dconvolution_dimension_numbers_test.cc102 ConvolutionDimensionNumbers dim_nums = in XLA_TEST_F() local
/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/lhlo_gpu_to_tfrt_gpu/
Dgemm_pattern.cc221 const mlir::mhlo::DotDimensionNumbers dim_nums = in GemmOpConversionRewrite() local
/external/tensorflow/tensorflow/compiler/mlir/xla/experimental/conv_emitter/
Dconv_emitter.cc534 const auto& dim_nums = conv->convolution_dimension_numbers(); in EmitConvolutionForwardAsMlir() local