Searched defs:element_shapes (Results 1 – 6 of 6) sorted by relevance
541 std::vector<const Shape*> element_shapes; in MakeTuple() local555 std::vector<const Shape*> element_shapes; in MakeTupleFromSlices() local569 std::vector<const Shape*> element_shapes; in MakeTupleOwned() local
1750 std::vector<const Shape*> element_shapes; in MoveIntoTuple() local
415 std::vector<Shape> element_shapes(kElementCount, element_shape); in XLA_TEST_F() local457 std::vector<Shape> element_shapes(kFanout, element_shape); in XLA_TEST_F() local
1270 std::vector<TensorShape> element_shapes(num_tensors, tensor_value->shape()); in Compute() local
742 absl::InlinedVector<const Shape*, 2> element_shapes; in MaybeMakeTupleShape() local
1917 std::vector<const Shape*> element_shapes; in CreateTuple() local