Home
last modified time | relevance | path

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

/external/pytorch/torch/csrc/jit/codegen/fuser/cuda/
Dfused_kernel.cpp90 std::vector<PartitionDesc> chunk_desc, in FusedKernelCUDA()
251 std::vector<PartitionDesc> chunk_desc, in createFusionKernel()
/external/pytorch/torch/csrc/jit/codegen/fuser/cpu/
Dfused_kernel.cpp311 std::vector<PartitionDesc> chunk_desc, in FusedKernelCPU()
346 std::vector<PartitionDesc> chunk_desc, in createFusionKernel()
/external/pytorch/torch/csrc/jit/codegen/fuser/
Dcompiler.cpp225 std::vector<PartitionDesc> chunk_desc; in compileKernel() local
Dexecutor.cpp36 auto& chunk_desc = spec.inputChunks().at(arg_idx); in getMapSize() local