| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | constant_folding.h | 42 static string AddControlDependency(const string& input_name, GraphDef* graph,
53 GraphDef* output) override;
56 const GraphDef& optimize_output, double result) override;
83 Status FoldMergeNode(NodeDef* node, GraphDef* output_graph);
84 Status FoldNode(NodeDef* node, GraphDef* output_graph,
91 NodeDef* node, GraphDef* graph);
94 NodeDef* node, GraphDef* graph);
95 void ReplaceSubtractionFromZeroByNegation(NodeDef* node, GraphDef* graph);
99 NodeDef* node, GraphDef* graph,
101 void ReplaceDivisionOfOnesByReciprocal(NodeDef* node, GraphDef* graph);
[all …]
|
| D | auto_parallel.h | 37 GraphDef* output) override;
40 const GraphDef& optimize_output, double result) override;
43 GraphDef graph_;
56 GraphDef* graph);
58 void AddSharedNodes(GraphDef* graph);
59 void AddOneReplica(GraphDef* graph, int number);
60 void BuildGraph(GraphDef* graph);
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | remote_fused_graph_execute_utils.h | 106 const GraphDef& graph_def,
117 const GraphDef& graph_def,
145 static bool GetOutputTensorShapeType(const GraphDef& graph_def,
150 const GraphDef& graph_def,
172 const bool dry_run_inference, GraphDef* graph_def);
176 const string& executor_name, const GraphDef& subgraph_def,
185 const GraphDef& subgraph_def, const std::vector<string>& inputs,
198 const GraphDef& graph_def,
203 const GraphDef& graph_def,
204 GraphDef* subgraph_def);
[all …]
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | transform_utils.h | 35 void MapNamesToNodes(const GraphDef& graph_def,
40 void MapNodesToOutputs(const GraphDef& graph_def,
108 void FilterGraphDef(const GraphDef& input_graph_def,
110 GraphDef* output_graph_def);
114 void RemoveAttributes(const GraphDef& input_graph_def,
116 GraphDef* output_graph_def);
121 Status SortByExecutionOrder(const GraphDef& input_graph_def,
122 GraphDef* output_graph_def);
125 void FindInvalidInputs(const GraphDef& graph_def,
130 Status IsGraphValid(const GraphDef& graph_def);
[all …]
|
| D | quantize_nodes_test.cc | 35 Status QuantizeNodes(const GraphDef& input_graph_def,
37 GraphDef* output_graph_def);
38 Status RemoveRedundantQuantizations(const GraphDef& input_graph_def,
40 GraphDef* output_graph_def);
41 Status QuantizePlaceholders(const GraphDef& input_graph_def,
43 GraphDef* output_graph_def);
44 Status ConvertFakeQuantsToRequantize(const GraphDef& input_graph_def,
46 GraphDef* output_graph_def);
47 Status MergeAdjacentRequantizes(const GraphDef& input_graph_def,
49 GraphDef* output_graph_def);
[all …]
|
| D | fuse_convolutions_test.cc | 32 Status FuseResizePadAndConv(const GraphDef& input_graph_def,
34 GraphDef* output_graph_def);
35 Status FuseResizeAndConv(const GraphDef& input_graph_def,
37 GraphDef* output_graph_def);
38 Status FusePadAndConv(const GraphDef& input_graph_def,
40 GraphDef* output_graph_def);
75 GraphDef original_graph_def; in TestFuseResizePadAndConv()
83 GraphDef fused_graph_def; in TestFuseResizePadAndConv()
125 GraphDef original_graph_def; in TestFuseResizeAndConv()
133 GraphDef fused_graph_def; in TestFuseResizeAndConv()
[all …]
|
| D | obfuscate_names_test.cc | 32 Status ObfuscateNames(const GraphDef& input_graph_def,
34 GraphDef* output_graph_def);
39 GraphDef graph_def; in TestSimpleTree()
75 GraphDef result; in TestSimpleTree()
92 GraphDef graph_def; in TestManyNodes()
99 GraphDef result; in TestManyNodes()
111 GraphDef graph_def; in TestNameClashes()
118 GraphDef result; in TestNameClashes()
|
| D | set_device_test.cc | 32 Status SetDevice(const GraphDef& input_graph_def,
34 GraphDef* output_graph_def);
37 GraphDef CreateDeviceGraph() { in CreateDeviceGraph()
38 GraphDef graph_def; in CreateDeviceGraph()
83 GraphDef graph_def = CreateDeviceGraph(); in TEST()
84 GraphDef result; in TEST()
104 GraphDef graph_def = CreateDeviceGraph(); in TEST()
105 GraphDef result; in TEST()
|
| D | backports_test.cc | 33 Status BackportConcatV2Transform(const GraphDef& input_graph_def,
35 GraphDef* output_graph_def);
36 Status BackportTensorArrayV3Transform(const GraphDef& input_graph_def,
38 GraphDef* output_graph_def);
43 GraphDef graph_def; in TestBackportConcatV2()
83 GraphDef result; in TestBackportConcatV2()
109 GraphDef graph_def; in TEST()
155 GraphDef result; in TEST()
181 GraphDef graph_def; in TEST()
186 GraphDef result; in TEST()
|
| D | fold_constants_test.cc | 38 Status ReplaceSendRecvs(const GraphDef& original_graph_def,
39 const GraphDef& rewritten_graph_def,
42 GraphDef* output_graph_def);
70 GraphDef graph_def; in TestSimpleAdd()
107 GraphDef graph_def; in TestOpExclusionAdd()
131 GraphDef graph_def; in TestShapePropagation()
160 GraphDef graph_def; in TestPreserveOutputShapes()
173 void TestConstantFolding(const GraphDef& graph_def, in TestConstantFolding()
184 GraphDef folded_graph_def; in TestConstantFolding()
235 GraphDef o_graph_def; in TestReplaceSendRecvs()
[all …]
|
| D | strip_unused_nodes_test.cc | 33 Status StripUnusedNodes(const GraphDef& input_graph_def,
35 GraphDef* output_graph_def);
40 GraphDef graph_def; in TestSimpleAdd()
59 GraphDef result; in TestSimpleAdd()
71 GraphDef graph_def; in TestCommonAncestor()
113 GraphDef result; in TestCommonAncestor()
131 GraphDef graph_def; in TestSimplePlaceholder()
156 GraphDef result; in TestSimplePlaceholder()
174 GraphDef graph_def; in TestPlaceholderDefaultArgs()
199 GraphDef result; in TestPlaceholderDefaultArgs()
[all …]
|
| D | sort_by_execution_order_test.cc | 33 void GetOrder(const GraphDef& graph_def, std::map<string, int>* order) { in GetOrder()
41 GraphDef graph_def; in TestSimpleAdd()
56 GraphDef result; in TestSimpleAdd()
67 GraphDef graph_def; in TestSimpleLinear()
88 GraphDef result; in TestSimpleLinear()
100 GraphDef graph_def; in TestSimpleTree()
136 GraphDef result; in TestSimpleTree()
151 GraphDef graph_def; in TestCommonAncestor()
183 GraphDef result; in TestCommonAncestor()
|
| D | fuse_convolutions.cc | 29 Status FuseResizePadAndConv(const GraphDef& input_graph_def, in FuseResizePadAndConv()
31 GraphDef* output_graph_def) { in FuseResizePadAndConv()
32 GraphDef replaced_graph_def; in FuseResizePadAndConv()
83 Status FuseResizeAndConv(const GraphDef& input_graph_def, in FuseResizeAndConv()
85 GraphDef* output_graph_def) { in FuseResizeAndConv()
86 GraphDef replaced_graph_def; in FuseResizeAndConv()
138 Status FusePadAndConv(const GraphDef& input_graph_def, in FusePadAndConv()
140 GraphDef* output_graph_def) { in FusePadAndConv()
141 GraphDef replaced_graph_def; in FusePadAndConv()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | graph_def_util.h | 26 class GraphDef; variable
30 string SummarizeGraphDef(const GraphDef& graph_def);
44 Status ValidateExternalGraphDefSyntax(const GraphDef& graph_def);
55 Status AddDefaultAttrsToGraphDef(GraphDef* graph_def,
61 Status AddDefaultAttrsToGraphDef(GraphDef* graph_def,
96 GraphDef* graph_def, const OpRegistryInterface& consumer_op_registry,
103 void OpsUsedByGraph(const GraphDef& graph_def,
115 Status StrippedOpListForGraph(const GraphDef& graph_def,
|
| D | graph_def_util_test.cc | 46 GraphDef graph_def; in TEST()
49 GraphDef expected_graph_def = graph_def; in TEST()
66 GraphDef graph_def; in TEST()
70 GraphDef expected_graph_def = graph_def; in TEST()
94 GraphDef produced_graph_def; in TEST()
103 GraphDef expected_graph_def; in TEST()
127 GraphDef produced_graph_def; in TEST()
132 GraphDef expected_graph_def = produced_graph_def; in TEST()
160 GraphDef produced_graph_def; in TEST()
164 GraphDef expected_graph_def = produced_graph_def; in TEST()
[all …]
|
| /external/tensorflow/tensorflow/core/grappler/graph_analyzer/ |
| D | test_tools.h | 84 GraphDef graph_3n_self_control_;
86 GraphDef graph_multi_input_;
88 GraphDef graph_all_or_none_;
90 GraphDef graph_circular_onedir_;
92 GraphDef graph_circular_bidir_;
94 GraphDef graph_linear_;
96 GraphDef graph_cross_;
97 GraphDef graph_small_cross_;
100 GraphDef graph_for_link_order_;
102 GraphDef graph_sun_;
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | graph_utils_test.cc | 41 GraphDef graph_def; in TEST()
49 GraphDef graph_def; in TEST()
57 GraphDef graph_def; in TEST()
65 GraphDef graph_def; in TEST()
73 GraphDef graph_def; in TEST()
81 GraphDef graph_def; in TEST()
89 GraphDef graph_def_a; in TEST()
91 GraphDef graph_def_b; in TEST()
105 GraphDef graph_def; in TEST()
127 GraphDef graph_def; in TEST()
[all …]
|
| D | graph_utils.h | 83 bool Compare(const GraphDef& g1, const GraphDef& g2);
86 bool ContainsGraphNodeWithName(StringPiece name, const GraphDef& graph);
93 bool ContainsNodeWithOp(StringPiece op, const GraphDef& graph);
97 int FindGraphNodeWithName(StringPiece name, const GraphDef& graph);
106 int FindGraphNodeWithOp(StringPiece op, const GraphDef& graph);
118 const GraphDef& graph);
122 void SetUniqueGraphNodeName(StringPiece prefix, GraphDef* graph, NodeDef* node);
148 Status FindSinkNode(const GraphDef& graph_def, NodeDef* sink_node);
|
| /external/tensorflow/tensorflow/lite/toco/tensorflow_graph_matching/ |
| D | resolve_cluster.cc | 32 using tensorflow::GraphDef;
36 const std::vector<string>& cluster_names, GraphDef* graph) { in AddNodeToGraph()
64 const GraphDef& graph_def, in FindCluster()
87 std::unique_ptr<GraphDef> MaybeResolveClusters( in MaybeResolveClusters()
88 const GraphDef& graph_def, in MaybeResolveClusters()
90 std::unique_ptr<GraphDef> pruned_graph(new GraphDef); in MaybeResolveClusters()
136 std::unique_ptr<GraphDef> MaybeReplaceCompositeSubgraph( in MaybeReplaceCompositeSubgraph()
137 const GraphDef& tf_graph) { in MaybeReplaceCompositeSubgraph()
143 std::unique_ptr<GraphDef> pruned_graph = in MaybeReplaceCompositeSubgraph()
|
| D | resolve_cluster.h | 34 std::unique_ptr<tensorflow::GraphDef> MaybeResolveClusters(
35 const tensorflow::GraphDef& graph_def,
44 tensorflow::GraphDef* graph);
51 const tensorflow::GraphDef& graph_def,
58 std::unique_ptr<tensorflow::GraphDef> MaybeReplaceCompositeSubgraph(
59 const tensorflow::GraphDef& tf_graph);
|
| /external/tensorflow/tensorflow/cc/tools/ |
| D | freeze_saved_model_test.cc | 34 void GraphDefEqual(const GraphDef& actual, const GraphDef& expected) { in GraphDefEqual()
62 const GraphDef& graph_def, const string& init_node, in InitializeSavedModelBundleSession()
77 Status AddGraphDefToSavedModelBundle(const GraphDef& graph_def, in AddGraphDefToSavedModelBundle()
89 const GraphDef& graph_def, const std::unordered_set<string>& outputs, in AddGraphDefWithOutputsToSavedModelBundle()
102 const GraphDef& frozen_graph_def, in RunAndCompareFrozenAndUnfrozenGraphs()
123 GraphDef graph_def; in TestFreezeGraphWithoutDependentVariables()
145 GraphDef frozen_graph_def; in TestFreezeGraphWithoutDependentVariables()
151 GraphDef expected_graph_def; in TestFreezeGraphWithoutDependentVariables()
169 GraphDef graph_def; in TestFreezeGraphWithDependentVariables()
190 GraphDef frozen_graph_def; in TestFreezeGraphWithDependentVariables()
[all …]
|
| /external/tensorflow/tensorflow/core/grappler/ |
| D | mutable_graph_view_test.cc | 145 GraphDef graph_def = test::function::GDef( in TEST()
156 GraphDef subgraph = test::function::GDef( in TEST()
173 GraphDef graph_def; in TEST()
177 GraphDef subgraph = test::function::GDef({}, {x_times_two}); in TEST()
186 GraphDef graph_def = test::function::GDef({}, {x_times_two}); in TEST()
189 GraphDef subgraph = test::function::GDef({}, {x_times_two}); in TEST()
199 GraphDef graph_def = test::function::GDef({}, {x_times_four}); in TEST()
203 GraphDef subgraph = test::function::GDef({}, {x_times_two}); in TEST()
214 GraphDef graph_def = test::function::GDef( in TEST()
242 GraphDef graph_def = test::function::GDef( in TEST()
[all …]
|
| D | graph_topology_view.h | 58 Status InitializeFromGraph(const GraphDef& graph,
61 Status InitializeFromGraph(const GraphDef& graph,
63 Status InitializeFromGraph(const GraphDef& graph, bool ignore_control_edges);
64 Status InitializeFromGraph(const GraphDef& graph);
68 const GraphDef* graph() const { return graph_; } in graph()
99 const GraphDef* graph_ = nullptr; // do not own
|
| /external/tensorflow/tensorflow/core/lib/strings/ |
| D | proto_serialization_test.cc | 30 GraphDef MakeGraphDef(int num_nodes) { in MakeGraphDef()
31 GraphDef graph_def; in MakeGraphDef()
45 GraphDef graph_def = MakeGraphDef(num_nodes); in BM_ProtoSerializationToString()
58 GraphDef graph_def = MakeGraphDef(num_nodes); in BM_ProtoSerializationToBuffer()
72 GraphDef graph_def = MakeGraphDef(num_nodes); in BM_DeterministicProtoHash64()
83 GraphDef graph_def_a = MakeGraphDef(num_nodes); in BM_AreSerializedProtosEqual()
84 GraphDef graph_def_b = MakeGraphDef(num_nodes); in BM_AreSerializedProtosEqual()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_fusion_optimizer_test.cc | 34 std::unordered_map<string, string> GetClusters(const GraphDef& graph) { in GetClusters()
48 GraphDef graph; in TEST_F()
65 GraphDef output; in TEST_F()
77 GraphDef graph; in TEST_F()
97 GraphDef output; in TEST_F()
106 GraphDef graph; in TEST_F()
130 GraphDef output; in TEST_F()
137 GraphDef graph; in TEST_F()
154 GraphDef output; in TEST_F()
161 GraphDef graph; in TEST_F()
[all …]
|