Searched refs:b_batch_dims (Results 1 – 2 of 2) sorted by relevance
143 ShapeHandle b_batch_dims; in __anon57f5a5dd0202() local146 TF_RETURN_IF_ERROR(c->Subshape(b_shape, 0, -2, &b_batch_dims)); in __anon57f5a5dd0202()147 TF_RETURN_IF_ERROR(c->Merge(a_batch_dims, b_batch_dims, &batch_dims)); in __anon57f5a5dd0202()
1936 absl::Span<const int64> b_batch_dims(b.dimensions()); in InferTriangularSolveShape() local1938 b_batch_dims.remove_suffix(2); in InferTriangularSolveShape()1939 if (a_batch_dims != b_batch_dims) { in InferTriangularSolveShape()