Searched defs:num_thread (Results 1 – 6 of 6) sorted by relevance
255 void StrideTest(TfLiteRegistration* registration, int num_thread) { in StrideTest()293 void PaddingTest(TfLiteRegistration* registration, int num_thread) { in PaddingTest()332 void SimpleTest(TfLiteRegistration* registration, int num_thread) { in SimpleTest()368 int num_thread) { in SimpleDilatedTestPaddingValid()430 int num_thread) { in SimpleDilatedTestPaddingSame()474 void BatchPaddingValidTest(TfLiteRegistration* registration, int num_thread) { in BatchPaddingValidTest()535 void BatchPaddingSameTest(TfLiteRegistration* registration, int num_thread) { in BatchPaddingSameTest()
368 int64_t num_thread = in cpu_flash_attention() local371 int64_t num_thread = 1; in cpu_flash_attention() local
91 const int num_closure = 100, num_thread = 2; in TEST() local
244 int64_t num_thread = at::get_num_threads(); in cpu_flash_attention() local499 int64_t num_thread = at::get_num_threads(); in cpu_flash_attention_backward() local
279 int num_thread) in map_and_thread()
519 util_queue_finish_execute(void *data, void *gdata, int num_thread) in util_queue_finish_execute()