Home
last modified time | relevance | path

Searched refs:rhs_contracting_dimension (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_evaluator_typed_visitor.h1231 const int64 rhs_contracting_dimension = dnums.rhs_contracting_dimensions(0); in HandleDot() local
1234 rhs->shape().dimensions(rhs_contracting_dimension)) in HandleDot()
1238 << rhs->shape().dimensions(rhs_contracting_dimension); in HandleDot()
1242 rhs_contracting_dimension == 0 && in HandleDot()
Dshape_inference.cc648 const int64 rhs_contracting_dimension = in InferDotOpShape() local
651 rhs.dimensions(rhs_contracting_dimension) || in InferDotOpShape()
653 rhs.is_dynamic_dimension(rhs_contracting_dimension)) { in InferDotOpShape()
Dalgebraic_simplifier.cc1551 int rhs_contracting_dimension = dnums.rhs_contracting_dimensions(0); in OptimizeDotOfGather() local
1556 rhs, lhs, /*a_contracting_dimension=*/rhs_contracting_dimension)) { in OptimizeDotOfGather()
1591 right_operand->shape().dimensions(1 - rhs_contracting_dimension); in OptimizeDotOfGather()
1601 : 1 - rhs_contracting_dimension; in OptimizeDotOfGather()