Home
last modified time | relevance | path

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

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