Home
last modified time | relevance | path

Searched refs:optimized_module (Results 1 – 11 of 11) sorted by relevance

/aosp_15_r20/external/pytorch/test/mobile/model_test/
H A Dtorchvision_models.py14 optimized_module = optimize_for_mobile(traced_script_module)
16 optimized_module,
21 optimized_module(example)
22 return optimized_module
31 optimized_module = optimize_for_mobile(traced_script_module, backend="vulkan")
33 optimized_module,
38 optimized_module(example)
39 return optimized_module
48 optimized_module = optimize_for_mobile(traced_script_module)
50 optimized_module,
[all …]
/aosp_15_r20/external/pytorch/test/inductor/
H A Dtest_config.py127 optimized_module = torch.compile(
130 compiler_config = optimized_module.get_compiler_config()
135 optimized_module = torch.compile(model)
136 compiler_config = optimized_module.get_compiler_config()
143 optimized_module = torch.compile(
146 compiler_config = optimized_module.get_compiler_config()
150 optimized_module = torch.compile(dummy_fn)
151 compiler_config = optimized_module.get_compiler_config()
158 optimized_module = torch.compile(dummy_fn, backend="eager")
159 compiler_config = optimized_module.get_compiler_config()
H A Dtest_torchinductor.py11236 def get_triton_codegen(optimized_module, args): argument
11238 result = optimized_module(*args)
/aosp_15_r20/external/pytorch/binaries/
H A Doptimize_for_mobile.cc85 torch::jit::Module optimized_module; in main() local
87 optimized_module = torch::jit::optimizeForMobile(module); in main()
89 optimized_module = torch::jit::vulkanOptimizeForMobile( in main()
92 optimized_module = torch::jit::metalOptimizeForMobile(module, preserved_methods); in main()
96 auto new_ops = torch::jit::export_opnames(optimized_module); in main()
104 optimized_module._save_for_mobile(output_model_name); in main()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
H A Dreduction_vectorization_test.cc79 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<VerifiedHloModule> optimized_module, in TEST_F()
95 CompileAndOptionallyVerifyPtx(std::move(optimized_module), expected); in TEST_F()
146 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<VerifiedHloModule> optimized_module, in TEST_F()
148 CompileAndOptionallyVerifyPtx(std::move(optimized_module), expected_ptx); in TEST_F()
H A Dgemm_rewrite_test.cc47 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> optimized_module, in CheckNumberOfAllocations()
52 std::move(optimized_module), backend().default_stream_executor(), in CheckNumberOfAllocations()
139 std::unique_ptr<HloModule> optimized_module,
146 RunFileCheck(optimized_module->ToString(),
156 RunFileCheck(optimized_module->ToString(),
H A Dgpu_compilation_parallelism_test.cc55 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<VerifiedHloModule> optimized_module, in TEST_F()
61 CompileAndOptionallyVerifyPtx(std::move(optimized_module), in TEST_F()
H A Dgpu_copy_alone_test.cc48 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<VerifiedHloModule> optimized_module, in TEST_F()
53 CompileAndOptionallyVerifyPtx(std::move(optimized_module), in TEST_F()
H A Dgpu_too_many_blocks_test.cc45 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> optimized_module, in TEST_F()
50 std::move(optimized_module), backend().default_stream_executor(), in TEST_F()
H A Dgpu_spmd_e2e_compile_test.cc74 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> optimized_module, in TEST_F()
80 optimized_module->entry_computation()->instructions(), in TEST_F()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dllvm_irgen_test_base.cc89 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> optimized_module, in MatchOptimizedHlo()
94 RunFileCheck(optimized_module->ToString(print_opts), pattern); in MatchOptimizedHlo()