Searched refs:element_shapes (Results 1 – 6 of 6) sorted by relevance
541 std::vector<const Shape*> element_shapes; in MakeTuple() local542 element_shapes.reserve(elements.size()); in MakeTuple()544 element_shapes.push_back(&element->shape()); in MakeTuple()546 Literal literal(ShapeUtil::MakeTupleShapeWithPtrs(element_shapes)); in MakeTuple()555 std::vector<const Shape*> element_shapes; in MakeTupleFromSlices() local556 element_shapes.reserve(elements.size()); in MakeTupleFromSlices()558 element_shapes.push_back(&element.shape()); in MakeTupleFromSlices()560 Literal literal(ShapeUtil::MakeTupleShapeWithPtrs(element_shapes)); in MakeTupleFromSlices()569 std::vector<const Shape*> element_shapes; in MakeTupleOwned() local570 element_shapes.reserve(elements.size()); in MakeTupleOwned()[all …]
1750 std::vector<const Shape*> element_shapes; in MoveIntoTuple() local1751 element_shapes.reserve(elements.size()); in MoveIntoTuple()1753 element_shapes.push_back(&element.shape()); in MoveIntoTuple()1755 Literal literal(ShapeUtil::MakeTupleShapeWithPtrs(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()457 std::vector<Shape> element_shapes(kFanout, element_shape); in XLA_TEST_F() local458 const Shape inner_tuple_shape = ShapeUtil::MakeTupleShape(element_shapes); in XLA_TEST_F()
742 absl::InlinedVector<const Shape*, 2> element_shapes; in MaybeMakeTupleShape() local743 element_shapes.reserve(phlos.size()); in MaybeMakeTupleShape()745 element_shapes.push_back(&phlo.hlo()->shape()); in MaybeMakeTupleShape()747 return ShapeUtil::MakeTupleShapeWithPtrs(element_shapes); in MaybeMakeTupleShape()
1270 std::vector<TensorShape> element_shapes(num_tensors, tensor_value->shape()); in Compute() local1272 element_shapes[i].set_dim(0, tensor_lengths_t(i)); in Compute()1311 ctx, ctx->allocate_temp(tensor_array->ElemType(), element_shapes[i], in Compute()
1917 std::vector<const Shape*> element_shapes; in CreateTuple() local1918 element_shapes.reserve(elements.size()); in CreateTuple()1920 element_shapes.push_back(&element->shape()); in CreateTuple()1922 Shape tuple_shape = ShapeUtil::MakeTupleShapeWithPtrs(element_shapes); in CreateTuple()