Home
last modified time | relevance | path

Searched refs:lhs_contracting_dimension (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/ops/
Dxla_ops.cc198 const int64 lhs_contracting_dimension = in __anonbbc1c2d10202() local
204 c->DimKnownRank(lhs_shape_handle, lhs_contracting_dimension)); in __anonbbc1c2d10202()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_evaluator_typed_visitor.h1350 const int64 lhs_contracting_dimension = dnums.lhs_contracting_dimensions(0);
1353 CHECK_EQ(lhs->shape().dimensions(lhs_contracting_dimension),
1356 << lhs->shape().dimensions(lhs_contracting_dimension)
1361 if (lhs_rank == 2 && rhs_rank == 2 && lhs_contracting_dimension == 1 &&
1372 lhs->shape().dimensions(lhs_contracting_dimension);
Dalgebraic_simplifier.cc2004 int lhs_contracting_dimension = dnums.lhs_contracting_dimensions(0); in OptimizeDotOfGather() local
2008 lhs, rhs, /*a_contracting_dimension=*/lhs_contracting_dimension) && in OptimizeDotOfGather()
2043 const int m = left_operand->shape().dimensions(1 - lhs_contracting_dimension); in OptimizeDotOfGather()
2055 ? 1 - lhs_contracting_dimension in OptimizeDotOfGather()
Dshape_inference.cc678 const int64 lhs_contracting_dimension = in InferDotOpShape() local
682 if (lhs.dimensions(lhs_contracting_dimension) != in InferDotOpShape()