Searched refs:input_bcast_shapes (Results 1 – 1 of 1) sorted by relevance
180 gtl::InlinedVector<ShapeHandle, 2> input_bcast_shapes(c->num_inputs()); in EinsumShape() local203 input_bcast_shapes[i] = c->Scalar(); in EinsumShape()221 input_bcast_shapes[i] = c->UnknownShape(); in EinsumShape()227 &input_bcast_shapes[i])); in EinsumShape()255 if (input_bcast_shapes.size() == 1) { in EinsumShape()256 output_bcast_shape = input_bcast_shapes[0]; in EinsumShape()257 } else if (input_bcast_shapes.size() == 2) { in EinsumShape()259 c, input_bcast_shapes[0], input_bcast_shapes[1], true, in EinsumShape()