Home
last modified time | relevance | path

Searched defs:lhs_contracting_dims (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Ddot_handler.cc1513 std::vector<int64> lhs_contracting_dims; in PartitionBaseCase() local
1623 std::vector<int64> lhs_contracting_dims; in PartitionBaseCase() local
1825 std::vector<int64> lhs_contracting_dims; in PartitionDotGroupOnBatch() local
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dlegalize_to_linalg.cc1206 auto lhs_contracting_dims = in matchAndRewrite() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_parser.cc2380 optional<std::vector<int64>> lhs_contracting_dims; in ParseInstructionRhs() local
Dalgebraic_simplifier.cc2108 auto lhs_contracting_dims = dnums.lhs_contracting_dimensions(); in OptimizeDotOfReorderContractingDims() local