Searched refs:optimize_for_inference (Results 1 – 16 of 16) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/tools/ |
| H A D | BUILD | 19 ":optimize_for_inference",
226 name = "optimize_for_inference",
227 srcs = ["optimize_for_inference.py"],
235 srcs = ["optimize_for_inference.py"],
|
| H A D | optimize_for_inference.py | 83 output_graph_def = optimize_for_inference_lib.optimize_for_inference(
|
| H A D | optimize_for_inference_lib.py | 89 def optimize_for_inference(input_graph_def, input_node_names, output_node_names, function
|
| H A D | optimize_for_inference_test.py | 123 output = optimize_for_inference_lib.optimize_for_inference(
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/api/ |
| H A D | module.cpp | 493 void optimize_for_inference(std::shared_ptr<Graph> graph) { in optimize_for_inference() function
500 Module optimize_for_inference( in optimize_for_inference() function
511 optimize_for_inference(frozen_mod.get_method("forward").graph()); in optimize_for_inference()
514 optimize_for_inference(frozen_mod.get_method(method).graph()); in optimize_for_inference()
|
| H A D | module.h | 344 TORCH_API Module optimize_for_inference(
|
| /aosp_15_r20/external/pytorch/test/cpp/jit/ |
| H A D | test_module_api.cpp | 402 auto frozen_mod2 = torch::jit::optimize_for_inference(m); in TEST()
424 auto frozen_mod = torch::jit::optimize_for_inference(m); in TEST()
440 torch::jit::optimize_for_inference(m, std::vector<std::string>{"bar"}); in TEST()
|
| /aosp_15_r20/external/pytorch/docs/source/ |
| H A D | torch.compiler_custom_backends.rst | 248 with `optimize_for_inference <https://pytorch.org/docs/stable/generated/torch.jit.optimize_for_infe…
254 return torch.jit.optimize_for_inference(scripted)
|
| H A D | jit.rst | 63 optimize_for_inference
|
| /aosp_15_r20/external/pytorch/torch/jit/ |
| H A D | _freeze.py | 182 def optimize_for_inference( function
|
| H A D | __init__.py | 26 from torch.jit._freeze import freeze, optimize_for_inference, run_frozen_optimizations
|
| /aosp_15_r20/external/pytorch/torch/_functorch/ |
| H A D | compilers.py | 95 f = torch.jit.optimize_for_inference(f)
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_fx_experimental.py | 1523 optimized_model = optimization.optimize_for_inference(model)
1526 optimized_model2 = optimization.optimize_for_inference(
1557 optimized_model = optimization.optimize_for_inference(model)
|
| /aosp_15_r20/external/pytorch/torch/fx/experimental/ |
| H A D | optimization.py | 240 def optimize_for_inference( function
|
| /aosp_15_r20/external/pytorch/test/jit/ |
| H A D | test_freezing.py | 3014 frozen_mod = torch.jit.optimize_for_inference(scripted_mod)
3068 optimized_mod = torch.jit.optimize_for_inference(
3113 exported = torch.jit.optimize_for_inference(exported)
3327 optimized = torch.jit.optimize_for_inference(scripted)
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/python/ |
| H A D | init.cpp | 424 optimize_for_inference(module, other_methods); in initJITBindings()
|