Searched refs:element_shapes (Results 1 – 5 of 5) sorted by relevance
470 std::vector<Shape> element_shapes; in MakeTuple() local472 element_shapes.push_back(element->shape()); in MakeTuple()474 Literal literal(ShapeUtil::MakeTupleShape(element_shapes)); in MakeTuple()483 std::vector<Shape> element_shapes; in MakeTupleFromSlices() local485 element_shapes.push_back(element.shape()); in MakeTupleFromSlices()487 Literal literal(ShapeUtil::MakeTupleShape(element_shapes)); in MakeTupleFromSlices()496 std::vector<Shape> element_shapes; in MakeTupleOwned() local497 element_shapes.reserve(elements.size()); in MakeTupleOwned()499 element_shapes.push_back(element.shape()); in MakeTupleOwned()501 Literal literal(ShapeUtil::MakeTupleShape(element_shapes)); in MakeTupleOwned()
1552 std::vector<Shape> element_shapes; in MoveIntoTuple() local1554 element_shapes.push_back(element.shape()); in MoveIntoTuple()1556 Literal literal(ShapeUtil::MakeTupleShape(element_shapes), in MoveIntoTuple()
415 std::vector<Shape> element_shapes(kElementCount, element_shape); in XLA_TEST_F() local416 const Shape tuple_shape = ShapeUtil::MakeTupleShape(element_shapes); in XLA_TEST_F()455 std::vector<Shape> element_shapes(kFanout, element_shape); in XLA_TEST_F() local456 const Shape inner_tuple_shape = ShapeUtil::MakeTupleShape(element_shapes); in XLA_TEST_F()
1278 std::vector<TensorShape> element_shapes(num_tensors, tensor_value->shape()); in Compute() local1280 element_shapes[i].set_dim(0, tensor_lengths_t(i)); in Compute()1320 tensor_array->ElemType(), element_shapes[i], in Compute()
1607 std::vector<Shape> element_shapes; in CreateTuple() local1609 element_shapes.push_back(element->shape()); in CreateTuple()1611 Shape tuple_shape = ShapeUtil::MakeTupleShape(element_shapes); in CreateTuple()