Home
last modified time | relevance | path

Searched defs:queue_callback (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/batching_util/
H A Dadaptive_shared_batch_scheduler_test.cc107 auto queue_callback = [&mu, &processed_batches, &finish_processing]( in TEST() local
150 auto queue_callback = [&env](std::unique_ptr<Batch<FakeTask>> batch) { in TEST() local
217 auto queue_callback = [&mu, &processed_batches, &finish_processing]( in TEST() local
292 auto queue_callback = [&mu, &processed_batches, &finish_processing]( in TEST() local
354 auto queue_callback = [&mu, &processed_batches, &finish_processing]( in TEST() local
391 auto queue_callback = [&mu, &processed_batches, &finish_processing]( in TEST() local
432 auto queue_callback = [](std::unique_ptr<Batch<FakeTask>> batch) { in TEST() local
447 auto queue_callback = in TEST() local
488 auto queue_callback = in TEST() local
H A Dserial_device_batch_scheduler_test.cc101 auto queue_callback = [&mu, &processed_batches, &finish_processing]( in TEST() local
150 auto queue_callback = [&mu, &processed_batches, &start_processing, &pending, in TEST() local
210 auto queue_callback = in TEST() local
268 auto queue_callback = [&mu, &processed_batches, &finish_processing]( in TEST() local
318 auto queue_callback = in TEST() local
356 auto queue_callback = [&mu, &processed_batches, &finish_processing]( in TEST() local
/aosp_15_r20/external/pytorch/torch/_dynamo/
H A Dexternal_utils.py114 def queue_callback(final_callbacks, cb): member in FakeCompiledAutogradEngine
/aosp_15_r20/external/pytorch/torch/csrc/autograd/
H A Dengine.cpp1410 void Engine::queue_callback(std::function<void()> callback) { in queue_callback() function in torch::autograd::Engine