Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/binaries/
H A Dbenchmark_args.h20 C10_DEFINE_string(
26 C10_DEFINE_string(init_net, "", "The given net to initialize any parameters.");
27 C10_DEFINE_string(
32 C10_DEFINE_string(
40 C10_DEFINE_string(
47 C10_DEFINE_string(
58 C10_DEFINE_string(net, "", "The given net to benchmark.");
59 C10_DEFINE_string(
66 C10_DEFINE_string(
H A Daot_model_compiler.cc17 C10_DEFINE_string(model, "", "The torch script model to optimize.");
18 C10_DEFINE_string(model_name, "", "The name of the model.");
19 C10_DEFINE_string(model_version, "", "The version of the model.");
20 C10_DEFINE_string(
26 C10_DEFINE_string(
33 C10_DEFINE_string(
39 C10_DEFINE_string(
43 C10_DEFINE_string(method_name, "forward", "The name of the method.");
44 C10_DEFINE_string(
48 C10_DEFINE_string(output_model, "", "Name of the output model to be saved.");
H A Dcompare_models_torch.cc30 C10_DEFINE_string(
34 C10_DEFINE_string(
38 C10_DEFINE_string(
46 C10_DEFINE_string(input_type, "", "Input type (uint8_t/float)");
47 C10_DEFINE_string(
68 C10_DEFINE_string(
72 C10_DEFINE_string(
76 C10_DEFINE_string(tolerance, "1e-5", "tolerance to use for comparison");
H A Doptimize_for_mobile.cc27 C10_DEFINE_string(model, "", "The torch script model to optimize.");
28 C10_DEFINE_string(
32 C10_DEFINE_string(backend, "", "The backend to be optimized");
33 C10_DEFINE_string(preserved_methods, "", "Methods to be preserved")
H A Dspeed_benchmark_torch.cc34 C10_DEFINE_string(model, "", "The given torch script model to benchmark.");
35 C10_DEFINE_string(
43 C10_DEFINE_string(input_type, "", "Input type (uint8_t/float)");
44 C10_DEFINE_string(
H A Ddump_operator_names.cc55 C10_DEFINE_string(model, "", "The given torch script model.");
56 C10_DEFINE_string(output, "", "The output yaml file of operator list.");
H A Dlite_interpreter_model_load.cc9 C10_DEFINE_string(model, "", "The given bytecode model to check if it is supported by lite_interpre…
H A Dload_benchmark_torch.cc34 C10_DEFINE_string(model, "", "The given torch script model to benchmark.");
/aosp_15_r20/external/pytorch/torch/csrc/jit/mobile/model_tracer/
H A Dtracer.cpp34 C10_DEFINE_string(
39 C10_DEFINE_string(
/aosp_15_r20/external/pytorch/c10/util/
H A DFlags.h130 #define C10_DEFINE_string(name, default_value, help_str) \ macro
207 #define C10_DEFINE_string(name, default_value, help_str) \ macro
/aosp_15_r20/external/pytorch/torch/csrc/lazy/core/
H A Dconfig.cpp64 C10_DEFINE_string(