| /external/tensorflow/tensorflow/contrib/quantize/python/ |
| D | quantize_graph.py | 26 def _create_graph(input_graph=None, argument
63 if input_graph is None:
64 input_graph = ops.get_default_graph()
68 _check_for_training_ops(input_graph)
69 with input_graph.as_default():
71 input_graph,
75 input_graph,
84 def create_training_graph(input_graph=None, quant_delay=0): argument
119 input_graph=input_graph,
125 def create_eval_graph(input_graph=None): argument
[all …]
|
| /external/tensorflow/tensorflow/python/tools/ |
| D | freeze_graph.py | 242 def _parse_input_graph_proto(input_graph, input_binary): argument
244 if not gfile.Exists(input_graph):
245 print("Input graph file '" + input_graph + "' does not exist!")
249 with gfile.GFile(input_graph, mode) as f:
257 def _parse_input_meta_graph_proto(input_graph, input_binary): argument
259 if not gfile.Exists(input_graph):
260 print("Input meta graph file '" + input_graph + "' does not exist!")
264 with gfile.GFile(input_graph, mode) as f:
269 print("Loaded meta graph file '" + input_graph)
288 def freeze_graph(input_graph, argument
[all …]
|
| D | strip_unused_lib.py | 90 def strip_unused_from_files(input_graph, input_binary, output_graph, argument
95 if not gfile.Exists(input_graph):
96 print("Input graph file '" + input_graph + "' does not exist!")
105 with gfile.GFile(input_graph, mode) as f:
|
| D | strip_unused.py | 55 strip_unused_lib.strip_unused_from_files(FLAGS.input_graph,
|
| /external/tensorflow/tensorflow/python/training/ |
| D | quantize_training.i | 23 const string& input_graph, in DoQuantizeTrainingOnGraphDefHelper() argument
29 tensorflow::DoQuantizeTrainingOnSerializedGraphDef(input_graph, num_bits, in DoQuantizeTrainingOnGraphDefHelper()
52 const string& input_graph,
65 def do_quantize_training_on_graphdef(input_graph, num_bits):
76 input_graph.SerializeToString(), num_bits, status)
|
| /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/contrib/model_pruning/python/ |
| D | strip_pruning_vars_lib.py | 55 input_graph = ops.Graph()
56 with input_graph.as_default():
59 with session.Session(graph=input_graph) as sess:
|
| D | strip_pruning_vars_test.py | 135 def _get_outputs(self, session, input_graph, tensors_list, graph_prefix=None): argument
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | concatenate_dataset_op.cc | 98 Node* input_graph = nullptr; in AsGraphDefInternal() local
99 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph)); in AsGraphDefInternal()
104 b->AddDataset(this, {input_graph, to_concatenate_graph}, output)); in AsGraphDefInternal()
|
| D | cache_dataset_ops.cc | 94 Node* input_graph = nullptr; in AsGraphDefInternal() local
95 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_graph)); in AsGraphDefInternal()
98 TF_RETURN_IF_ERROR(b->AddDataset(this, {input_graph, filename}, output)); in AsGraphDefInternal()
|
| /external/tensorflow/tensorflow/core/graph/ |
| 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 | 297 argspec: "args=[\'input_graph\', \'num_bits\'], varargs=None, keywords=None, defaults=None"
|
| /external/tensorflow/tensorflow/contrib/quantize/ |
| D | README.md | 42 tf.contrib.quantize.create_training_graph(input_graph=g,
61 tf.contrib.quantize.create_eval_graph(input_graph=g)
88 --input_graph=eval_graph_def.pb \
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | graph_util_impl.py | 311 def remove_training_nodes(input_graph, protected_nodes=None): argument
335 input_nodes = input_graph.node
|
| /external/tensorflow/tensorflow/lite/python/ |
| D | create_custom_op.py | 101 graph = _read_graph_def(filename=flags.FLAGS.input_graph)
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | model_pruner.cc | 400 Status SetTransitiveFaninGraph(const GraphDef& input_graph, in SetTransitiveFaninGraph() argument
407 ComputeTransitiveFanin(input_graph, terminal_nodes, &ill_formed); in SetTransitiveFaninGraph()
|
| /external/tensorflow/tensorflow/lite/g3doc/guide/ |
| D | get_started.md | 117 freeze_graph --input_graph=/tmp/mobilenet_v1_224.pb \
127 Set `input_graph` and `input_checkpoint` to the respective filenames. The
|
| /external/tensorflow/tensorflow/compiler/aot/ |
| D | tfcompile.bzl | 131 " --input_graph=$(location " + graph + ")" +
|