Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/
Dpod_tpu_driver.cc402 std::vector<int64_t> children_ids; in AllocateTuple() local
407 children_ids.push_back(child_op_id); in AllocateTuple()
412 [this, core_id, region, children_ids, in AllocateTuple()
416 child_buffers.reserve(children_ids.size()); in AllocateTuple()
417 for (int i = 0; i < children_ids.size(); ++i) { in AllocateTuple()
418 CHECK_EXISTS_OR_RETURN(underlying_buffers_, children_ids[i], in AllocateTuple()
420 child_buffers.push_back(underlying_buffers_[children_ids[i]].get()); in AllocateTuple()