Searched refs:input_graph_def (Results 1 - 25 of 66) sorted by relevance

123

/external/tensorflow/tensorflow/tools/graph_transforms/
H A Dsort_by_execution_order.cc34 const GraphDef& input_graph_def, const TransformFuncContext& unused_context,
36 return SortByExecutionOrder(input_graph_def, output_graph_def);
33 SortByExecutionOrderWithUnusedContext( const GraphDef& input_graph_def, const TransformFuncContext& unused_context, GraphDef* output_graph_def) argument
H A Dadd_default_attributes.cc24 Status AddDefaultAttributes(const GraphDef& input_graph_def, argument
30 input_graph_def.library()));
32 *output_graph_def = input_graph_def;
H A Dfake_quantize_training.cc36 Status FakeQuantizeTraining(const GraphDef& input_graph_def, argument
44 return DoQuantizeTrainingOnGraphDef(input_graph_def, num_bits, quant_op_type,
H A Dfold_constants_lib.h30 Status FoldConstants(const GraphDef& input_graph_def,
36 Status RemoveUnusedNodes(const GraphDef& input_graph_def,
H A Dremove_device.cc31 Status RemoveDevice(const GraphDef& input_graph_def, argument
35 for (const NodeDef& node : input_graph_def.node()) {
H A Dremove_control_dependencies.cc26 Status RemoveControlDependencies(const GraphDef& input_graph_def, argument
30 for (const NodeDef& node : input_graph_def.node()) {
H A Dset_device.cc23 Status SetDevice(const GraphDef& input_graph_def, argument
33 for (const NodeDef& node : input_graph_def.node()) {
H A D__init__.py27 def TransformGraph(input_graph_def, inputs, outputs, transforms):
35 input_graph_def: GraphDef object containing a model to be transformed.
44 input_graph_def_string = input_graph_def.SerializeToString()
H A Dremove_attribute.cc31 Status RemoveAttribute(const GraphDef& input_graph_def, argument
55 for (const NodeDef& node : input_graph_def.node()) {
H A Drename_op.cc31 Status RenameOp(const GraphDef& input_graph_def, argument
46 for (const NodeDef& node : input_graph_def.node()) {
H A Dobfuscate_names.cc32 Status ObfuscateNames(const GraphDef& input_graph_def, argument
49 for (const NodeDef& input_node : input_graph_def.node()) {
74 for (const NodeDef& input_node : input_graph_def.node()) {
H A Dbackports.cc31 Status BackportConcatV2Transform(const GraphDef& input_graph_def, argument
35 input_graph_def, {"ConcatV2"},
65 Status BackportTensorArrayV3Transform(const GraphDef& input_graph_def, argument
71 input_graph_def, {"TensorArrayV3|TensorArrayGradV3"},
H A Dremove_ema.cc28 Status RemoveEMA(const GraphDef& input_graph_def, argument
32 input_graph_def, // clang-format off
H A Drename_attribute.cc30 Status RenameAttribute(const GraphDef& input_graph_def, argument
53 for (const NodeDef& node : input_graph_def.node()) {
H A Dremove_ema_test.cc28 Status FakeQuantizeTraining(const GraphDef& input_graph_def,
32 Status RemoveEMA(const GraphDef& input_graph_def,
36 Status QuantizeNodes(const GraphDef& input_graph_def,
H A Dfake_quantize_training_test.cc27 Status FakeQuantizeTraining(const GraphDef& input_graph_def,
H A Dfold_constants_lib.cc120 Status RemoveUnusedNodes(const GraphDef& input_graph_def, argument
124 for (const NodeDef& node : input_graph_def.node()) {
165 input_graph_def,
188 Status FoldConstants(const GraphDef& input_graph_def, argument
192 TF_RETURN_IF_ERROR(input_graph.AddFunctionLibrary(input_graph_def.library()));
207 RemoveAttributes(input_graph_def, {"_output_shapes"}, &cleaned_graph_def);
213 ImportGraphDef({}, input_graph_def, &input_graph, &shape_refiner));
293 TF_RETURN_IF_ERROR(ReplaceSendRecvs(input_graph_def, folded_graph_def,
H A Dfuse_convolutions.cc30 Status FuseResizePadAndConv(const GraphDef& input_graph_def, argument
35 input_graph_def, // clang-format off
84 Status FuseResizeAndConv(const GraphDef& input_graph_def, argument
89 input_graph_def, // clang-format off
139 Status FusePadAndConv(const GraphDef& input_graph_def, argument
144 input_graph_def, // clang-format off
/external/tensorflow/tensorflow/tools/graph_transforms/python/
H A Dtransform_graph_test.py35 input_graph_def = graph_pb2.GraphDef()
37 const_op1 = input_graph_def.node.add()
46 const_op2 = input_graph_def.node.add()
56 add_op = input_graph_def.node.add()
64 relu_op = input_graph_def.node.add()
76 transformed_graph_def = TransformGraph(input_graph_def, input_names,
/external/tensorflow/tensorflow/python/tools/
H A Dstrip_unused_lib.py32 def strip_unused(input_graph_def, input_node_names, output_node_names,
37 input_graph_def: A graph with nodes we want to prune.
61 for node in input_graph_def.node:
103 input_graph_def = graph_pb2.GraphDef()
107 input_graph_def.ParseFromString(f.read())
109 text_format.Merge(f.read(), input_graph_def)
111 output_graph_def = strip_unused(input_graph_def,
H A Dfreeze_graph.py62 def freeze_graph_with_def_protos(input_graph_def,
96 elif input_graph_def:
97 for node in input_graph_def.node:
100 if input_graph_def:
101 _ = importer.import_graph_def(input_graph_def, name="")
152 input_graph_def,
170 input_graph_def = graph_pb2.GraphDef()
174 input_graph_def.ParseFromString(f.read())
176 text_format.Merge(f.read(), input_graph_def)
177 return input_graph_def
[all...]
H A Doptimize_for_inference.py79 input_graph_def = graph_pb2.GraphDef()
83 input_graph_def.ParseFromString(data)
85 text_format.Merge(data.decode("utf-8"), input_graph_def)
88 input_graph_def,
H A Doptimize_for_inference_lib.py89 def optimize_for_inference(input_graph_def, input_node_names, output_node_names,
94 input_graph_def: A GraphDef containing a training model.
105 ensure_graph_is_valid(input_graph_def)
106 optimized_graph_def = input_graph_def
201 def fold_batch_norms(input_graph_def):
215 input_graph_def: A GraphDef containing a model.
224 for node in input_graph_def.node:
232 for node in input_graph_def.node:
357 for node in input_graph_def.node:
368 def fuse_resize_and_conv(input_graph_def, output_node_name
[all...]
/external/tensorflow/tensorflow/contrib/tensorrt/python/
H A Dtrt_convert.py32 def create_inference_graph(input_graph_def,
40 input_graph_def: GraphDef object containing a model to be transformed.
78 input_graph_def_str = input_graph_def.SerializeToString()
/external/tensorflow/tensorflow/core/kernels/hexagon/
H A Dhexagon_rewriter_transform.cc44 const GraphDef& input_graph_def, const TransformFuncContext& context,
78 GraphDef mutable_input_graph_def = input_graph_def;
43 RewriteQuantizedStrippedModelForHexagon( const GraphDef& input_graph_def, const TransformFuncContext& context, GraphDef* output_graph_def) argument

Completed in 1775 milliseconds

123