Home
last modified time | relevance | path

Searched refs:sub_literals (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/client/
Dvalue_inference.cc46 std::vector<Literal> sub_literals; in CreatePredLiteral() local
48 sub_literals.reserve(reference_shape_tuple_shapes.size()); in CreatePredLiteral()
50 sub_literals.emplace_back(CreatePredLiteral(pred, shape)); in CreatePredLiteral()
52 return Literal::MoveIntoTuple(absl::MakeSpan(sub_literals)); in CreatePredLiteral()
67 std::vector<Literal> sub_literals; in CreateS64Literal() local
69 sub_literals.reserve(reference_shape_tuple_shapes.size()); in CreateS64Literal()
71 sub_literals.emplace_back(CreateS64Literal(value, shape)); in CreateS64Literal()
73 return Literal::MoveIntoTuple(absl::MakeSpan(sub_literals)); in CreateS64Literal()
89 std::vector<Literal> sub_literals; in CreateGarbageLiteral() local
91 sub_literals.emplace_back(CreateGarbageLiteral(shape)); in CreateGarbageLiteral()
[all …]