Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2xla/ops/
Dxla_ops.cc263 const int64_t lhs_batch_dimension = in XlaDotShapeFunction() local
269 c->Merge(c->DimKnownRank(lhs_shape_handle, lhs_batch_dimension), in XlaDotShapeFunction()
272 lhs_batch_dimension, " and rhs dimension ", rhs_batch_dimension); in XlaDotShapeFunction()