Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/framework/
Dcommon_shape_fns.cc330 ShapeHandle b_batch_shape; in BatchMatMulV2Shape() local
333 TF_RETURN_IF_ERROR(c->Subshape(b_shape, 0, -2, &b_batch_shape)); in BatchMatMulV2Shape()
336 c, a_batch_shape, b_batch_shape, true, &output_batch_shape)); in BatchMatMulV2Shape()