Searched refs:reference_module (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tools/ |
| H A D | prepare_reference_module.cc | 49 std::unique_ptr<HloModule> reference_module = in PrepareReferenceModule() local
53 module_modifier_hook(test_module, test_runner, reference_module.get())); in PrepareReferenceModule()
55 TF_RETURN_IF_ERROR(Despecializer().Run(reference_module.get()).status()); in PrepareReferenceModule()
57 return std::move(reference_module); in PrepareReferenceModule()
|
| H A D | run_hlo_module.cc | 178 std::unique_ptr<HloModule> reference_module; in RunAndCompare() local
182 reference_module = in RunAndCompare()
200 if (reference_module == nullptr) { in RunAndCompare()
206 ExecuteWithRunner(std::move(reference_module), args, reference_runner, in RunAndCompare()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/ |
| H A D | hlo_test_base.cc | 294 std::unique_ptr<HloModule> reference_module = test_module.Clone(); in MakeReferenceModule() local
298 reference_preprocessor(reference_module.get()); in MakeReferenceModule()
300 GetProgramShapeWithLayout(*reference_module))) { in MakeReferenceModule()
305 TF_RETURN_IF_ERROR(hlo_verifier_->Run(reference_module.get()).status()); in MakeReferenceModule()
306 return std::move(reference_module); in MakeReferenceModule()
315 TF_ASSIGN_OR_RETURN(auto reference_module, in RunAndCompareInternal()
323 reference_runner_.Execute(std::move(reference_module), in RunAndCompareInternal()
|
| /aosp_15_r20/external/pytorch/torch/onnx/_internal/fx/passes/ |
| H A D | _utils.py | 81 module: torch.fx.GraphModule, reference_module: torch.fx.GraphModule
98 node for node in reference_module.graph.nodes if node.op == "placeholder"
|
| H A D | modularization.py | 843 reference_module = torch.fx.GraphModule(self.module, self.module.graph)
845 reference_module,
|