Searched refs:result_parts (Results 1 – 1 of 1) sorted by relevance
181 std::vector<xla::XlaOp> result_parts; in SetTensorListPushIndex() local182 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() local388 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 …]