Searched refs:WaitForDefinitionEventOnStream (Results 1 – 6 of 6) sorted by relevance
77 void WaitForDefinitionEventOnStream(se::Stream* stream);
72 void XlaTensor::WaitForDefinitionEventOnStream(se::Stream* stream) { in WaitForDefinitionEventOnStream() function in tensorflow::XlaTensor
240 xla_tensor->WaitForDefinitionEventOnStream(device_to_host_stream.get()); in CopyDeviceTensorToCPU()
216 xla_input->WaitForDefinitionEventOnStream(dst_device_to_device_stream); in TpuDeviceToDeviceCopy()
264 xla_tensor->WaitForDefinitionEventOnStream( in PopulateInputs()
333 xla_tensor->WaitForDefinitionEventOnStream(stream); in BuildComputationInputs()