Searched refs:loaded_module (Results 1 – 8 of 8) sorted by relevance
| /external/pytorch/test/jit/ |
| D | test_generator.py | 169 loaded_module = torch.jit.load(saved_module)
171 self.assertEqual(eager_foo.foo.weight, loaded_module.foo.weight)
172 self.assertEqual(eager_foo.bar.weight, loaded_module.bar.weight)
180 out2, r2 = loaded_module(x)
194 print(loaded_module.forward.code)
|
| D | test_save_load.py | 632 loaded_module = torch.jit.load(path, _restore_shapes=True)
633 print(loaded_module.graph)
634 return traced_inputs, list(loaded_module.graph.inputs())
647 loaded_module = torch.jit.load(path, _restore_shapes=True)
648 loaded_inputs = list(loaded_module.graph.inputs())
654 loaded_module = torch.jit.load(path)
655 loaded_inputs = list(loaded_module.graph.inputs())
666 loaded_module = torch.jit.load(path, _restore_shapes=True)
667 loaded_inputs = list(loaded_module.graph.inputs())
670 loaded_module = torch.jit.load(path)
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | nvptx_compiler.cc | 259 std::unique_ptr<llvm::Module> loaded_module = in MaybeLoadLLVMFromFile() local
262 if (!loaded_module) { in MaybeLoadLLVMFromFile()
267 llvm_ir::DumpIrIfEnabled(*module, *loaded_module, /*optimized=*/false); in MaybeLoadLLVMFromFile()
268 return loaded_module; in MaybeLoadLLVMFromFile()
347 std::unique_ptr<llvm::Module> loaded_module = in CompileTargetBinary() local
350 if (loaded_module) { in CompileTargetBinary()
351 selected_module = loaded_module.get(); in CompileTargetBinary()
|
| /external/pytorch/test/mobile/nnc/ |
| D | test_nnc_backend.cpp | 96 auto loaded_module = _load_for_mobile(ss); in TEST() local
97 auto result = loaded_module.forward(inputs); in TEST()
|
| /external/pytorch/test/cpp/jit/ |
| D | test_custom_class.cpp | 59 auto loaded_module = torch::jit::load(iss, torch::kCPU); in TEST() local
140 auto loaded_module = torch::jit::load(iss, torch::kCPU); in TEST() local
|
| D | test_save_load.cpp | 76 auto loaded_module = torch::jit::load(iss, torch::kCPU, loaded_extra_files); in TEST() local
|
| /external/pytorch/docs/source/ |
| D | package.rst | 332 loaded_module = PackageImporter(my_package).import_module("foo.bar")
333 loaded_module.is_in_package() # True
|
| /external/pytorch/test/quantization/core/ |
| D | test_workflow_ops.py | 676 loaded_module = torch.jit.load(buf)
677 self.assertEqual(fq_module.calculate_qparams(), loaded_module.calculate_qparams())
|