Searched refs:ExtCallMemoryReuse (Results 1 – 2 of 2) sorted by relevance
19 ExtCallMemoryReuse extCallMemoryReuse(buffer_args_); in CodeGen()236 std::unordered_map<std::string, std::string> ExtCallMemoryReuse::256 ExtCallMemoryReuse::extCallFuncNameMap_ = makeExtCallFuncNameMap();258 ExtCallMemoryReuse::ExtCallMemoryReuse( in ExtCallMemoryReuse() function in torch::jit::tensorexpr::ExtCallMemoryReuse267 StmtPtr ExtCallMemoryReuse::mutate(const ExternalCallPtr& v) { in mutate()
109 class TORCH_API ExtCallMemoryReuse : public IRMutator {114 explicit ExtCallMemoryReuse(116 ~ExtCallMemoryReuse() override = default;