Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_event_mgr_test.cc160 em.ThenDeleteTensors(stream.get(), v); in TEST()
179 em.ThenDeleteTensors(stream.get(), v); in TEST()
196 em.ThenDeleteTensors(stream1.get(), v1); in TEST()
201 em.ThenDeleteTensors(stream2.get(), v2); in TEST()
219 em.ThenDeleteTensors(stream.get(), v); in TEST()
Dgpu_event_mgr.h72 void ThenDeleteTensors(se::Stream* stream,
Dgpu_event_mgr.cc163 void EventMgr::ThenDeleteTensors(se::Stream* stream, in ThenDeleteTensors() function in tensorflow::EventMgr
Dgpu_device.cc608 em_->ThenDeleteTensors(stream, tensor_refs); in ConsumeListOfAccessedTensors()