Lines Matching refs:shapes
24 // Given shapes of two tensors, computes the broadcast shape.
34 gtl::InlinedVector<BCast::Vec, 4> shapes;
44 shapes.push_back(vec);
46 BCast bcast(shapes[0], shapes[1]);
49 "Incompatible shapes: [", str_util::Join(shapes[0], ","),
50 "] vs. [", str_util::Join(shapes[1], ","), "]"));
69 // Given shapes of two tensors, computes the reduction indices for the
83 gtl::InlinedVector<BCast::Vec, 4> shapes;
93 shapes.push_back(vec);
95 BCast bcast(shapes[0], shapes[1]);
98 "Incompatible shapes: [", str_util::Join(shapes[0], ","),
99 "] vs. [", str_util::Join(shapes[1], ","), "]"));
