Home
last modified time | relevance | path

Searched defs:dot_dimension_numbers (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dbackend_configs.proto59 xla.DotDimensionNumbers dot_dimension_numbers = 7; field
/external/tensorflow/tensorflow/compiler/xla/tests/
Dtest_utils.cc724 DotDimensionNumbers dot_dimension_numbers; in CreateCanonicalDot() local
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dlhlo_legalize_to_affine.cc63 mhlo::DotDimensionNumbers dot_dimension_numbers = in matchAndRewrite() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo.proto180 xla.DotDimensionNumbers dot_dimension_numbers = 30; field
Dhlo_instructions.h1879 const DotDimensionNumbers& dot_dimension_numbers() const { in dot_dimension_numbers() function
Dalgebraic_simplifier.cc2557 DotDimensionNumbers dot_dimension_numbers; in HandleDot() local
5818 DotDimensionNumbers dot_dimension_numbers; in SimplifyConvToDot() local
Dhlo_instruction.cc4553 const DotDimensionNumbers& HloInstruction::dot_dimension_numbers() const { in dot_dimension_numbers() function in xla::HloInstruction
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dlegalize_hlo.cc747 DotDimensionNumbers dot_dimension_numbers, in ConvertDot()
842 auto dot_dimension_numbers = DotDimensionNumbers::get( in ConvertDotOp() local
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dmlir_hlo_to_hlo.cc256 xla::DotDimensionNumbers dot_dimension_numbers; in Convert_dot_dimension_numbers() local
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/
Dhlo_ops.cc248 auto dot_dimension_numbers = op.dot_dimension_numbers(); in Verify() local