Home
last modified time | relevance | path

Searched refs:QueueTensors (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_event_mgr_test.cc49 void QueueTensors(se::Stream* stream, TensorReferenceVector* tensors) { in QueueTensors() function in tensorflow::TEST_EventMgrHelper
51 em_->QueueTensors(stream, tensors); in QueueTensors()
128 th.QueueTensors(stream.get(), v); in TEST()
139 th.QueueTensors(stream.get(), v); in TEST()
241 th.QueueTensors(stream.get(), v); in TEST()
Dgpu_event_mgr.h155 void QueueTensors(se::Stream* stream, TensorReferenceVector* tensors) in QueueTensors() function
Dgpu_event_mgr.cc186 QueueTensors(accumulated_stream_, accumulated_tensors_); in FlushAccumulatedTensors()