Searched defs:element_shapes (Results 1 – 5 of 5) sorted by relevance
418 std::vector<Shape> element_shapes; in MakeTuple() local431 std::vector<Shape> element_shapes; in MakeTupleFromSlices() local444 std::vector<Shape> element_shapes; in MakeTupleOwned() local
1447 std::vector<Shape> element_shapes; in MoveIntoTuple() local
412 std::vector<Shape> element_shapes(kElementCount, element_shape); in XLA_TEST_F() local452 std::vector<Shape> element_shapes(kFanout, element_shape); in XLA_TEST_F() local
1286 std::vector<TensorShape> element_shapes(num_tensors, tensor_value->shape()); in Compute() local
1298 std::vector<Shape> element_shapes; in CreateTuple() local