Searched defs:prepareRunArgs (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/csrc/jit/codegen/onednn/ | ||
| H A D | kernel.cpp | 125 std::tuple<RunArgs, RunArgs> LlgaKernel::prepareRunArgs( in prepareRunArgs() function in torch::jit::fuser::onednn::LlgaKernel |
| /aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/ | ||
| H A D | kernel.cpp | 1956 std::vector<CodeGen::CallArg> TensorExprKernel::prepareRunArgs( in prepareRunArgs() function in TensorExprKernel |