| /external/tensorflow/tensorflow/python/tools/ |
| D | freeze_graph.py | 239 def _parse_input_graph_proto(input_graph, input_binary): argument
241 if not gfile.Exists(input_graph):
242 raise IOError("Input graph file '" + input_graph + "' does not exist!")
245 with gfile.GFile(input_graph, mode) as f:
253 def _parse_input_meta_graph_proto(input_graph, input_binary): argument
255 if not gfile.Exists(input_graph):
256 raise IOError("Input meta graph file '" + input_graph + "' does not exist!")
259 with gfile.GFile(input_graph, mode) as f:
264 print("Loaded meta graph file '" + input_graph)
282 def freeze_graph(input_graph, argument
[all …]
|
| D | strip_unused_lib.py | 89 def strip_unused_from_files(input_graph, input_binary, output_graph, argument
94 if not gfile.Exists(input_graph):
95 print("Input graph file '" + input_graph + "' does not exist!")
104 with gfile.GFile(input_graph, mode) as f:
|
| D | strip_unused.py | 52 strip_unused_lib.strip_unused_from_files(FLAGS.input_graph,
|
| /external/tensorflow/tensorflow/python/training/ |
| D | quantize_training_wrapper.cc | 24 static PyObject* DoQuantizeTrainingOnGraphDefHelper(const string& input_graph, in DoQuantizeTrainingOnGraphDefHelper() argument
30 input_graph, num_bits, "QuantizeAndDequantizeV2", &result)); in DoQuantizeTrainingOnGraphDefHelper()
43 [](const py::object input_graph, int num_bits) { in PYBIND11_MODULE() argument
46 input_graph.cast<std::string>(), num_bits)); in PYBIND11_MODULE()
|
| D | quantize_training.py | 27 def do_quantize_training_on_graphdef(input_graph, num_bits): argument
43 input_graph.SerializeToString(), num_bits)
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | fold_constants_lib.cc | 216 Graph input_graph(OpRegistry::Global()); in FoldConstants() local
217 TF_RETURN_IF_ERROR(input_graph.AddFunctionLibrary(input_graph_def.library())); in FoldConstants()
219 ShapeRefiner shape_refiner(input_graph.versions(), input_graph.op_registry()); in FoldConstants()
223 &input_graph.flib_def()); in FoldConstants()
235 ImportGraphDef({}, cleaned_graph_def, &input_graph, &shape_refiner)); in FoldConstants()
238 ImportGraphDef({}, input_graph_def, &input_graph, &shape_refiner)); in FoldConstants()
256 for (const Node* const node : input_graph.nodes()) { in FoldConstants()
294 &input_graph, context.input_names, context.output_names, {}, {}, in FoldConstants()
324 &input_graph, &was_mutated)); in FoldConstants()
326 input_graph.ToGraphDef(&folded_graph_def); in FoldConstants()
|
| /external/tensorflow/tensorflow/core/data/service/ |
| D | auto_shard_rewriter.cc | 109 GraphDef input_graph = graph_def; in ApplyAutoShardRewrite() local
110 TF_ASSIGN_OR_RETURN(std::string dataset_node, GetDatasetNode(input_graph)); in ApplyAutoShardRewrite()
112 GetGrapplerItem(&input_graph, &dataset_node, /*add_fake_sinks=*/false); in ApplyAutoShardRewrite()
|
| /external/tensorflow/tensorflow/dtensor/cc/ |
| D | dtensor_graph_to_mlir_pass.cc | 71 Graph* input_graph = graph->get(); in RunOnGraph() local
88 module_ref = ConvertGraphToMlir(*input_graph, debug_info, *flib_def, in RunOnGraph()
|
| /external/tensorflow/tensorflow/core/grappler/utils/ |
| D | transitive_fanin.cc | 94 Status SetTransitiveFaninGraph(const GraphDef& input_graph, in SetTransitiveFaninGraph() argument
101 ComputeTransitiveFanin(input_graph, terminal_nodes, &keep)); in SetTransitiveFaninGraph()
|
| D | transitive_fanin.h | 43 Status SetTransitiveFaninGraph(const GraphDef& input_graph,
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | concatenate_dataset_op.cc | 142 Node* input_graph = nullptr; in AsGraphDefInternal() local
143 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph)); in AsGraphDefInternal()
148 b->AddDataset(this, {input_graph, to_concatenate_graph}, output)); in AsGraphDefInternal()
|
| D | cache_dataset_ops.cc | 688 Node* input_graph = nullptr; in AsGraphDefInternal() local
689 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph)); in AsGraphDefInternal()
692 TF_RETURN_IF_ERROR(b->AddDataset(this, {input_graph, filename}, output)); in AsGraphDefInternal()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | quantize_training.h | 43 Status DoQuantizeTrainingOnSerializedGraphDef(const string& input_graph,
|
| D | quantize_training_test.cc | 302 GraphDef input_graph; in TEST_F() local
303 graph->ToGraphDef(&input_graph); in TEST_F()
305 input_graph.SerializeToString(&input_string); in TEST_F()
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.graph_util.pbtxt | 21 …argspec: "args=[\'input_graph\', \'protected_nodes\'], varargs=None, keywords=None, defaults=[\'No…
|
| D | tensorflow.train.pbtxt | 305 argspec: "args=[\'input_graph\', \'num_bits\'], varargs=None, keywords=None, defaults=None"
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | graph_util_impl.py | 333 def remove_training_nodes(input_graph, protected_nodes=None): argument
357 input_nodes = input_graph.node
|
| D | convert_to_constants.py | 1111 input_graph = _GraphDef(converter_data.graph_def)
1114 input_graph.nodes[tensor_name].convert_variable_to_constant(
1117 converted_graph = input_graph.converted_self().graph_def
|
| /external/tensorflow/tensorflow/lite/g3doc/r1/convert/ |
| D | python_api.md | 167 --input_graph=/path/to/graph.pbtxt --input_binary=false
172 `--input_graph=/path/to/graph.pbtxt --input_binary=false` with
173 `--input_graph=/path/to/graph.pb`
|
| /external/tensorflow/tensorflow/compiler/aot/ |
| D | tfcompile.bzl | 244 " --input_graph=$(location " + graph + ")" +
|