Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dtensor_list_utils.cc181 std::vector<xla::XlaOp> result_parts; in SetTensorListPushIndex() local
182 result_parts.reserve(tuple_size); in SetTensorListPushIndex()
184 result_parts.push_back(xla::GetTupleElement(list, i)); in SetTensorListPushIndex()
186 result_parts.push_back(push_index); in SetTensorListPushIndex()
187 *result = xla::Tuple(list.builder(), result_parts); in SetTensorListPushIndex()
366 std::vector<xla::XlaOp> result_parts; in ExecuteTensorListPushBack() local
388 result_parts.push_back(updated_list_part); in ExecuteTensorListPushBack()
404 result_parts.push_back(updated_list_part); in ExecuteTensorListPushBack()
408 result_parts.push_back(updated_push_index); in ExecuteTensorListPushBack()
410 *result = xla::Tuple(b, result_parts); in ExecuteTensorListPushBack()
[all …]