Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/
Dpod_tpu_driver.cc528 auto src_op_id = static_cast<const PodBufferHandle*>(src)->operation_id(); in TransferFromDeviceToDevice() local
533 [this, operation_id, src_op_id, dst_op_id, dst_core_id]() in TransferFromDeviceToDevice()
535 CHECK_EXISTS_OR_RETURN(underlying_buffers_, src_op_id, in TransferFromDeviceToDevice()
540 auto src_iter = underlying_buffers_.find(src_op_id); in TransferFromDeviceToDevice()