| /external/tensorflow/tensorflow/core/kernels/ |
| D | remote_fused_graph_execute_utils.h | 97 const GraphDef& graph_def,
108 const GraphDef& graph_def,
136 static bool GetOutputTensorShapeType(const GraphDef& graph_def,
141 const GraphDef& graph_def,
163 const bool dry_run_inference, GraphDef* graph_def);
167 const string& executor_name, const GraphDef& subgraph_def,
176 const GraphDef& subgraph_def, const std::vector<string>& inputs,
189 const GraphDef& graph_def,
194 const GraphDef& graph_def,
195 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 | remove_ema_test.cc | 28 Status FakeQuantizeTraining(const GraphDef& input_graph_def,
30 GraphDef* output_graph_def);
32 Status RemoveEMA(const GraphDef& input_graph_def,
34 GraphDef* output_graph_def);
36 Status QuantizeNodes(const GraphDef& input_graph_def,
38 GraphDef* output_graph_def);
56 GraphDef graph_def; in TEST_F()
60 GraphDef fake_quantized_graph_def; in TEST_F()
77 GraphDef frozen_graph_def; in TEST_F()
98 GraphDef removed_ema_graph_def; in TEST_F()
[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 | fold_constants_test.cc | 37 Status ReplaceSendRecvs(const GraphDef& original_graph_def,
38 const GraphDef& rewritten_graph_def,
41 GraphDef* output_graph_def);
69 GraphDef graph_def; in TestSimpleAdd()
106 GraphDef graph_def; in TestOpExclusionAdd()
130 GraphDef graph_def; in TestShapePropagation()
159 GraphDef graph_def; in TestPreserveOutputShapes()
172 void TestConstantFolding(const GraphDef& graph_def, in TestConstantFolding()
183 GraphDef folded_graph_def; in TestConstantFolding()
234 GraphDef o_graph_def; in TestReplaceSendRecvs()
[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 | 32 Status BackportConcatV2Transform(const GraphDef& input_graph_def,
34 GraphDef* output_graph_def);
35 Status BackportTensorArrayV3Transform(const GraphDef& input_graph_def,
37 GraphDef* output_graph_def);
42 GraphDef graph_def; in TestBackportConcatV2()
82 GraphDef result; in TestBackportConcatV2()
108 GraphDef graph_def; in TEST()
154 GraphDef result; in TEST()
180 GraphDef graph_def; in TEST()
185 GraphDef result; in TEST()
|
| 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 | 30 Status FuseResizePadAndConv(const GraphDef& input_graph_def, in FuseResizePadAndConv()
32 GraphDef* output_graph_def) { in FuseResizePadAndConv()
33 GraphDef replaced_graph_def; in FuseResizePadAndConv()
84 Status FuseResizeAndConv(const GraphDef& input_graph_def, in FuseResizeAndConv()
86 GraphDef* output_graph_def) { in FuseResizeAndConv()
87 GraphDef replaced_graph_def; in FuseResizeAndConv()
139 Status FusePadAndConv(const GraphDef& input_graph_def, in FusePadAndConv()
141 GraphDef* output_graph_def) { in FusePadAndConv()
142 GraphDef replaced_graph_def; in FusePadAndConv()
|
| D | quantize_nodes.cc | 193 Status MergeDuplicateNodes(const GraphDef& input_graph_def, in MergeDuplicateNodes()
195 GraphDef* output_graph_def) { in MergeDuplicateNodes()
201 GraphDef current_graph_def = input_graph_def; in MergeDuplicateNodes()
221 GraphDef merged_graph_def; in MergeDuplicateNodes()
262 Status RemoveRedundantQuantizations(const GraphDef& input_graph_def, in RemoveRedundantQuantizations()
264 GraphDef* output_graph_def) { in RemoveRedundantQuantizations()
270 GraphDef replaced_graph_def; in RemoveRedundantQuantizations()
312 Status QuantizePlaceholders(const GraphDef& input_graph_def, in QuantizePlaceholders()
314 GraphDef* output_graph_def) { in QuantizePlaceholders()
327 GraphDef placeholder_graph_def; in QuantizePlaceholders()
[all …]
|
| D | quantize_weights_test.cc | 32 Status QuantizeWeights(const GraphDef& input_graph_def,
34 GraphDef* output_graph_def);
42 GraphDef* original_graph_def) { in BuildGraphDef()
62 GraphDef original_graph_def; in TestQuantizeWeights()
76 GraphDef quantized_graph_def; in TestQuantizeWeights()
118 GraphDef original_graph_def; in TEST_F()
131 GraphDef quantized_graph_def; in TEST_F()
|
| D | fold_constants_lib.h | 30 Status FoldConstants(const GraphDef& input_graph_def,
32 GraphDef* output_graph_def);
36 Status RemoveUnusedNodes(const GraphDef& input_graph_def,
38 GraphDef* output_graph_def);
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | constant_folding.h | 37 static string AddControlDependency(const string& input_name, GraphDef* graph,
48 GraphDef* output) override;
51 const GraphDef& optimize_output, double result) override;
76 Status FoldNode(NodeDef* node, GraphDef* output_graph);
81 GraphDef* graph);
83 GraphDef* graph);
86 NodeDef* node, GraphDef* graph);
87 void ReplaceDivisionOfOnesByReciprocal(NodeDef* node, GraphDef* graph);
88 Status FoldGraph(GraphDef* output);
93 Status SimplifyGraph(GraphDef* output, const GraphProperties& properties,
[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/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,
90 GraphDef* graph_def, const OpRegistryInterface& consumer_op_registry,
97 void OpsUsedByGraph(const GraphDef& graph_def,
109 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/cc/tools/ |
| D | freeze_saved_model_test.cc | 33 void GraphDefEqual(const GraphDef& actual, const GraphDef& expected) { in GraphDefEqual()
61 const GraphDef& graph_def, const string& init_node, in InitializeSavedModelBundleSession()
76 Status AddGraphDefToSavedModelBundle(const GraphDef& graph_def, in AddGraphDefToSavedModelBundle()
88 const GraphDef& graph_def, const std::unordered_set<string>& outputs, in AddGraphDefWithOutputsToSavedModelBundle()
101 const GraphDef& frozen_graph_def, in RunAndCompareFrozenAndUnfrozenGraphs()
128 GraphDef frozen_graph_def; in TEST_F()
147 GraphDef frozen_graph_def; in TEST_F()
162 GraphDef graph_def; in TEST_F()
169 GraphDef frozen_graph_def; in TEST_F()
181 GraphDef graph_def; in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/contrib/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/contrib/lite/toco/ |
| D | export_tensorflow.cc | 43 using tensorflow::GraphDef;
129 const GraphDef& tensorflow_graph) { in HasAlreadyExportedConst()
142 GraphDef* tensorflow_graph, in ConvertFloatTensorConst()
160 GraphDef* tensorflow_graph) { in ConvertFloatTensorConst()
176 GraphDef* tensorflow_graph) { in ConvertFloatTensorConst()
197 GraphDef* tensorflow_graph) { in ConvertFloatTensorConst()
218 GraphDef* tensorflow_graph) { in ConvertIntTensorConst()
242 GraphDef* tensorflow_graph) { in CreateIntTensorConst()
260 GraphDef* tensorflow_graph) { in CreateMatrixShapeTensorConst()
278 GraphDef* tensorflow_graph) { in CreateDummyConcatDimTensorConst()
[all …]
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | master_test.cc | 63 Status CreateSession(const GraphDef& def, string* handle, in CreateSession()
79 Status ExtendSession(const string& handle, const GraphDef& def, in ExtendSession()
140 GraphDef def; // Empty. in TEST_F()
159 GraphDef def; // Empty. in TEST_F()
170 GraphDef def_0; // Empty. in TEST_F()
183 GraphDef def_1; in TEST_F()
194 GraphDef def_2; in TEST_F()
211 GraphDef def_0; // Empty. in TEST_F()
218 GraphDef def_1; in TEST_F()
230 GraphDef def_0; // Empty. in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | graph_execution_state.h | 102 GraphDef* graph_def, const GraphExecutionStateOptions& options,
111 const GraphDef& original_graph_def,
130 Status Extend(const GraphDef& extension_def,
157 const GraphDef& original_graph_def() { return original_graph_def_; } in original_graph_def()
166 GraphExecutionState(GraphDef* graph_def,
183 GraphDef original_graph_def_; // Immutable after ctor.
|