Searched refs:WaitForDefinitionEventOnStream (Results 1 – 4 of 4) sorted by relevance
94 void WaitForDefinitionEventOnStream(se::Stream* stream);
75 void XlaTensor::WaitForDefinitionEventOnStream(se::Stream* stream) { in WaitForDefinitionEventOnStream() function in tensorflow::XlaTensor
232 xla_tensor->WaitForDefinitionEventOnStream(device_to_host_stream.get()); in CopyDeviceTensorToCPU()
235 xla_tensor->WaitForDefinitionEventOnStream(stream); in PopulateInputs()