/external/tensorflow/tensorflow/tools/graph_transforms/ |
H A D | sort_by_execution_order.cc | 34 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 D | add_default_attributes.cc | 24 Status AddDefaultAttributes(const GraphDef& input_graph_def, argument 30 input_graph_def.library())); 32 *output_graph_def = input_graph_def;
|
H A D | fake_quantize_training.cc | 36 Status FakeQuantizeTraining(const GraphDef& input_graph_def, argument 44 return DoQuantizeTrainingOnGraphDef(input_graph_def, num_bits, quant_op_type,
|
H A D | fold_constants_lib.h | 30 Status FoldConstants(const GraphDef& input_graph_def, 36 Status RemoveUnusedNodes(const GraphDef& input_graph_def,
|
H A D | remove_device.cc | 31 Status RemoveDevice(const GraphDef& input_graph_def, argument 35 for (const NodeDef& node : input_graph_def.node()) {
|
H A D | remove_control_dependencies.cc | 26 Status RemoveControlDependencies(const GraphDef& input_graph_def, argument 30 for (const NodeDef& node : input_graph_def.node()) {
|
H A D | set_device.cc | 23 Status SetDevice(const GraphDef& input_graph_def, argument 33 for (const NodeDef& node : input_graph_def.node()) {
|
H A D | __init__.py | 27 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 D | remove_attribute.cc | 31 Status RemoveAttribute(const GraphDef& input_graph_def, argument 55 for (const NodeDef& node : input_graph_def.node()) {
|
H A D | rename_op.cc | 31 Status RenameOp(const GraphDef& input_graph_def, argument 46 for (const NodeDef& node : input_graph_def.node()) {
|
H A D | obfuscate_names.cc | 32 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 D | backports.cc | 31 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 D | remove_ema.cc | 28 Status RemoveEMA(const GraphDef& input_graph_def, argument 32 input_graph_def, // clang-format off
|
H A D | rename_attribute.cc | 30 Status RenameAttribute(const GraphDef& input_graph_def, argument 53 for (const NodeDef& node : input_graph_def.node()) {
|
H A D | remove_ema_test.cc | 28 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 D | fake_quantize_training_test.cc | 27 Status FakeQuantizeTraining(const GraphDef& input_graph_def,
|
H A D | fold_constants_lib.cc | 120 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 D | fuse_convolutions.cc | 30 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 D | transform_graph_test.py | 35 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 D | strip_unused_lib.py | 32 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 D | freeze_graph.py | 62 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 D | optimize_for_inference.py | 79 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 D | optimize_for_inference_lib.py | 89 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 D | trt_convert.py | 32 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 D | hexagon_rewriter_transform.cc | 44 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
|