Home
last modified time | relevance | path

Searched refs:should_copy_tensor_memory (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/jit/mobile/
H A Dflatbuffer_loader.h62 bool should_copy_tensor_memory = false);
129 bool should_copy_tensor_memory);
H A Dflatbuffer_loader.cpp122 void setShouldCopyTensorMemory(bool should_copy_tensor_memory) { in setShouldCopyTensorMemory() argument
123 should_copy_tensor_memory_ = should_copy_tensor_memory; in setShouldCopyTensorMemory()
763 bool should_copy_tensor_memory) { in parse_and_initialize_mobile_module() argument
774 loader.setShouldCopyTensorMemory(should_copy_tensor_memory); in parse_and_initialize_mobile_module()
/aosp_15_r20/external/pytorch/test/cpp/jit/
H A Dtest_flatbuffer.cpp46 bool should_copy_tensor_memory = false) { in parse_mobile_module() argument
52 should_copy_tensor_memory); in parse_mobile_module()