Searched refs:compute_function (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/ |
| H A D | operator-run.c | 1370 pthreadpool_function_2d_t compute_function = NULL; in pytorch_qnnp_run_operator() local
1372 compute_function = in pytorch_qnnp_run_operator()
1377 compute_function = in pytorch_qnnp_run_operator()
1381 compute_function = in pytorch_qnnp_run_operator()
1389 compute_function, in pytorch_qnnp_run_operator()
1507 pthreadpool_function_1d_t compute_function = NULL; in pytorch_qnnp_run_operator() local
1509 compute_function = in pytorch_qnnp_run_operator()
1514 compute_function = in pytorch_qnnp_run_operator()
1518 compute_function = (pthreadpool_function_1d_t) in pytorch_qnnp_run_operator()
1525 threadpool, compute_function, &context, op->batch_size); in pytorch_qnnp_run_operator()
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/c/ |
| H A D | kernels.cc | 48 void (*compute_function)(void*, TF_OpKernelContext*); member
61 result->compute_function = compute_func; in TF_NewKernelBuilder()
184 builder_->compute_function, in Create()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| H A D | cpu_executable.h | 83 const ComputeFunctionType& compute_function() const { in compute_function() function
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/ |
| H A D | xla_jit_compiled_cpu_function.cc | 127 cpu_executable->compute_function(); in Compile()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| H A D | tfrt_cpu_pjrt_client.cc | 1489 cpu_executable->compute_function()(result_buffer, &run_options, nullptr, in ExecuteHelper()
1545 cpu_executable->compute_function()(result_buffer, &run_options, in ExecuteHelper()
|