Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dbackend_configs.proto57 xla.DotDimensionNumbers dot_dimension_numbers = 7; field
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_cse.cc167 const auto& dot_dimension_numbers = in AbslHashValue() local
Dhlo.proto200 xla.DotDimensionNumbers dot_dimension_numbers = 30; field
Dhlo_instructions.h2253 const DotDimensionNumbers& dot_dimension_numbers() const { in dot_dimension_numbers() function
Dhlo_instruction.cc4831 const DotDimensionNumbers& HloInstruction::dot_dimension_numbers() const { in dot_dimension_numbers() function in xla::HloInstruction
Dalgebraic_simplifier.cc6652 DotDimensionNumbers dot_dimension_numbers; in SimplifyConvToDot() local
/external/tensorflow/tensorflow/compiler/xla/tests/
Dtest_utils.cc779 DotDimensionNumbers dot_dimension_numbers; in CreateCanonicalDot() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dlegalize_hlo.cc1673 DotDimensionNumbersAttr dot_dimension_numbers, in ConvertDot()
1768 auto dot_dimension_numbers = in ConvertDotOp() local
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dmlir_hlo_to_hlo.cc352 xla::DotDimensionNumbers dot_dimension_numbers; in Convert_dot_dimension_numbers() local