Searched defs:call_with_numel (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/ | ||
| H A D | codegen.cpp | 99 void CodeGen::call_with_numel(void** args, int64_t numel) { in call_with_numel() function in torch::jit::tensorexpr::CodeGen |
| H A D | cuda_codegen.cpp | 1063 void CudaCodeGen::call_with_numel(void** args, int64_t numel) { in call_with_numel() function in torch::jit::tensorexpr::CudaCodeGen |
| H A D | llvm_codegen.cpp | 425 void LLVMCodeGen::call_with_numel(void** args, int64_t /* numel */) { in call_with_numel() function in LLVMCodeGen |