Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/batching_util/
Dbatch_scheduler_test.cc76 EXPECT_EQ(7, batch.RemoveTask()->size()); in TEST()
78 EXPECT_EQ(3, batch.RemoveTask()->size()); in TEST()
Dbatch_scheduler.h93 std::unique_ptr<TaskType> RemoveTask();
246 std::unique_ptr<TaskType> Batch<TaskType>::RemoveTask() { in RemoveTask() function
/external/tensorflow/tensorflow/core/data/service/
Ddispatcher_state.cc66 RemoveTask(update.remove_task()); in Apply()
188 void DispatcherState::RemoveTask(const RemoveTaskUpdate& remove_task) { in RemoveTask() function in tensorflow::data::DispatcherState
Ddispatcher_state.h267 void RemoveTask(const RemoveTaskUpdate& remove_task);