Home
last modified time | relevance | path

Searched defs:memory_plan (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/jit/mobile/nnc/
H A Dcontext.h163 const MemoryPlan& memory_plan() const { in memory_plan() function
167 void set_memory_plan(const MemoryPlan& memory_plan) { in set_memory_plan()
/aosp_15_r20/external/pytorch/test/mobile/nnc/
H A Dtest_context.cpp53 MemoryPlan memory_plan; in create_test_memory_plan() local
/aosp_15_r20/external/pytorch/torch/_inductor/codegen/
H A Dwrapper.py957 def memory_plan(self): member in WrapperCodeGen