Home
last modified time | relevance | path

Searched defs:input_graph (Results 1 – 13 of 13) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/tools/
H A Dfreeze_graph.py239 def _parse_input_graph_proto(input_graph, input_binary): argument
253 def _parse_input_meta_graph_proto(input_graph, input_binary): argument
282 def freeze_graph(input_graph, argument
H A Dstrip_unused_lib.py89 def strip_unused_from_files(input_graph, input_binary, output_graph, argument
/aosp_15_r20/external/tensorflow/tensorflow/python/training/
H A Dquantize_training_wrapper.cc24 static PyObject* DoQuantizeTrainingOnGraphDefHelper(const string& input_graph, in DoQuantizeTrainingOnGraphDefHelper()
43 [](const py::object input_graph, int num_bits) { in PYBIND11_MODULE()
H A Dquantize_training.py27 def do_quantize_training_on_graphdef(input_graph, num_bits): argument
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/cc/
H A Ddtensor_graph_to_mlir_pass.cc71 Graph* input_graph = graph->get(); in RunOnGraph() local
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/utils/
H A Dtransitive_fanin.cc94 Status SetTransitiveFaninGraph(const GraphDef& input_graph, in SetTransitiveFaninGraph()
/aosp_15_r20/external/tensorflow/tensorflow/core/data/service/
H A Dauto_shard_rewriter.cc109 GraphDef input_graph = graph_def; in ApplyAutoShardRewrite() local
/aosp_15_r20/external/tensorflow/tensorflow/python/framework/
H A Dgraph_util_impl.py333 def remove_training_nodes(input_graph, protected_nodes=None): argument
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/
H A Dconcatenate_dataset_op.cc142 Node* input_graph = nullptr; in AsGraphDefInternal() local
H A Dcache_dataset_ops.cc688 Node* input_graph = nullptr; in AsGraphDefInternal() local
/aosp_15_r20/external/tensorflow/tensorflow/tools/graph_transforms/
H A Dfold_constants_lib.cc216 Graph input_graph(OpRegistry::Global()); in FoldConstants() local
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dquantize_training_test.cc302 GraphDef input_graph; in TEST_F() local
/aosp_15_r20/external/pytorch/benchmarks/static_runtime/
H A Dtest_static_module.cc103 auto input_graph = std::make_shared<torch::jit::Graph>(); in TEST() local