Home
last modified time | relevance | path

Searched refs:AddModule (Results 1 – 14 of 14) sorted by relevance

/external/executorch/examples/models/toy_model/
D__init__.py7 from .model import AddModule, AddMulModule, LinearModule, MulModule, SoftmaxModule
10 AddModule,
Dmodel.py42 class AddModule(torch.nn.Module, EagerModelBase): class
/external/pytorch/test/jit/xnnpack/
Dtest_xnnpack_delegate.py99 class AddModule(torch.nn.Module): class
106 add_module = AddModule()
130 class AddModule(torch.nn.Module): class
134 add_module = AddModule()
/external/executorch/backends/apple/mps/test/
Dtest_mps_binary_ops.py65 class AddModule(torch.nn.Module): class
73 add_module = AddModule()
81 class AddModule(torch.nn.Module): class
89 add_module = AddModule()
114 class AddModule(torch.nn.Module): class
122 add_module = AddModule()
/external/pytorch/test/fx/
Dtest_pass_infra.py53 class AddModule(torch.nn.Module): class
66 m = AddModule()
107 m = AddModule()
199 m = AddModule()
Dtest_partitioner_order.py28 class AddModule(torch.nn.Module): class
38 m = AddModule()
/external/chromium-trace/catapult/common/py_vulcanize/py_vulcanize/
Dproject.py167 g.AddModule(m)
199 g.AddModule(cur)
219 def AddModule(self, m): member in _Graph
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dsimple_orc_jit.h80 llvm::Error AddModule(llvm::orc::ThreadSafeModule module);
Dsimple_orc_jit.cc235 llvm::Error SimpleOrcJIT::AddModule(llvm::orc::ThreadSafeModule module) { in AddModule() function in xla::cpu::SimpleOrcJIT
Dcpu_compiler.cc1269 cantFail((*jit)->AddModule(std::move(thread_safe_module))); in CompileLegacyCpuExecutable()
/external/executorch/examples/arm/
Daot_arm_compiler.py123 class AddModule(torch.nn.Module): class
176 "add": AddModule,
/external/llvm/bindings/go/llvm/
Dexecutionengine.go149 func (ee ExecutionEngine) AddModule(m Module) { C.LLVMAddModule(ee.C, m.C) } func
/external/executorch/backends/vulkan/test/
Dtest_vulkan_delegate.py185 class AddModule(torch.nn.Module): class
198 add_module = AddModule()
/external/pytorch/test/onnx/
Dtest_pytorch_onnx_onnxruntime.py3302 class AddModule(torch.nn.Module): class
3325 self.run_test(AddModule(), (x, y))
3334 self.run_test(AddModule(), (x, y))
3343 self.run_test(AddModule(), (x, y))
3352 self.run_test(AddModule(), (x, y))
3361 self.run_test(AddModule(), (x, y))