Searched refs:ResetDefinitionEvent (Results 1 – 6 of 6) sorted by relevance
85 void ResetDefinitionEvent(std::shared_ptr<se::Event> event,
90 void XlaTensor::ResetDefinitionEvent(std::shared_ptr<se::Event> event, in ResetDefinitionEvent() function in tensorflow::XlaTensor
176 xla_tensor->ResetDefinitionEvent(std::move(event), in CopyCPUTensorToDevice()
267 xla_output->ResetDefinitionEvent(std::move(definition_event), in TpuDeviceToDeviceCopy()
338 xla_tensor->ResetDefinitionEvent(definition_event, stream); in PopulateXlaTensor()
481 xla_tensor->ResetDefinitionEvent(definition_event, stream); in AllocateOutputTensors()