Searched refs:CanShapedBufferBeAccessedNow (Results 1 – 9 of 9) sorted by relevance
47 bool CanShapedBufferBeAccessedNow( in CanShapedBufferBeAccessedNow() function
73 bool CanShapedBufferBeAccessedNow(
38 if (transfer_manager->CanShapedBufferBeAccessedNow(stream->parent(), in PopulateResultTupleBuffers()
243 bool TpuTransferManager::CanShapedBufferBeAccessedNow( in CanShapedBufferBeAccessedNow() function in tensorflow::tpu::TpuTransferManager
268 virtual bool CanShapedBufferBeAccessedNow( in CanShapedBufferBeAccessedNow() function
162 !transfer_manager_->CanShapedBufferBeAccessedNow( in CopyCPUTensorToDevice()
220 if (!dst_xla_context->transfer_manager()->CanShapedBufferBeAccessedNow( in TpuDeviceToDeviceCopy()
692 if (transfer_manager->CanShapedBufferBeAccessedNow(stream->parent(), in DoWork()
387 DCHECK(transfer_manager->CanShapedBufferBeAccessedNow( in AllocateDestinationBuffer()430 DCHECK(transfer_manager->CanShapedBufferBeAccessedNow( in AllocateDestinationBuffer()