| /external/tensorflow/tensorflow/contrib/framework/python/framework/ |
| graph_util_test.py | 49 fused_graph_def = graph_util.fuse_op(
51 self.assertEqual(len(fused_graph_def.node), 4)
52 self.assertEqual(fused_graph_def.node[0].name, 'A')
53 self.assertEqual(fused_graph_def.node[1].name, 'FusedOp')
54 self.assertEqual(fused_graph_def.node[1].input[0], 'A')
55 self.assertEqual(fused_graph_def.node[1].op, 'Op2')
56 self.assertEqual(fused_graph_def.node[1].attr['_output_quantized'].b, True)
57 self.assertEqual(fused_graph_def.node[1].attr['_output_types'].list.type,
59 self.assertEqual(fused_graph_def.node[2].name, 'D')
60 self.assertEqual(fused_graph_def.node[3].name, 'E'
[all...] |
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| fuse_convolutions_test.cc | 83 GraphDef fused_graph_def; local
85 &fused_graph_def));
88 TF_ASSERT_OK(fused_session->Create(fused_graph_def));
94 for (const NodeDef& node : fused_graph_def.node()) {
133 GraphDef fused_graph_def; local
135 &fused_graph_def));
138 TF_ASSERT_OK(fused_session->Create(fused_graph_def));
144 for (const NodeDef& node : fused_graph_def.node()) {
185 GraphDef fused_graph_def; local
187 FusePadAndConv(original_graph_def, {{}, {"output"}}, &fused_graph_def));
[all...] |
| fold_batch_norms_test.cc | 74 GraphDef fused_graph_def; local
76 FoldBatchNorms(original_graph_def, {{}, {"output"}}, &fused_graph_def));
79 TF_ASSERT_OK(fused_session->Create(fused_graph_def));
85 for (const NodeDef& node : fused_graph_def.node()) {
134 GraphDef fused_graph_def; local
136 original_graph_def, {{}, {"output", "output_2"}}, &fused_graph_def));
139 TF_ASSERT_OK(fused_session->Create(fused_graph_def));
182 GraphDef fused_graph_def; local
184 FoldBatchNorms(original_graph_def, {{}, {"output"}}, &fused_graph_def));
187 TF_ASSERT_OK(fused_session->Create(fused_graph_def));
[all...] |
| fold_old_batch_norms_test.cc | 107 GraphDef fused_graph_def; local
109 &fused_graph_def));
112 TF_ASSERT_OK(fused_session->Create(fused_graph_def));
118 for (const NodeDef& node : fused_graph_def.node()) {
184 GraphDef fused_graph_def; local
186 &fused_graph_def));
189 TF_ASSERT_OK(fused_session->Create(fused_graph_def));
195 for (const NodeDef& node : fused_graph_def.node()) {
284 GraphDef fused_graph_def; local
286 &fused_graph_def));
[all...] |
| /external/tensorflow/tensorflow/core/kernels/ |
| i_remote_fused_graph_executor.h | 65 GraphDef* fused_graph_def) = 0;
|
| remote_fused_graph_execute_op_test_utils.h | 79 GraphDef* fused_graph_def) final;
|
| remote_fused_graph_execute_op_test.cc | 220 GraphDef* fused_graph_def) final {
221 *fused_graph_def = original_graph_def;
261 GraphDef fused_graph_def; local
262 TF_CHECK_OK(root.ToGraphDef(&fused_graph_def));
263 *fused_graph = fused_graph_def;
|
| remote_fused_graph_execute_op_test_utils.cc | 118 const std::vector<string>& outputs, GraphDef* fused_graph_def) {
122 /*require_shape_type=*/false, fused_graph_def);
|
| /external/tensorflow/tensorflow/core/kernels/hexagon/ |
| hexagon_control_wrapper.h | 54 GraphDef* fused_graph_def) final;
|
| hexagon_graph_execution_test.cc | 281 static void RunFusedGraph(const GraphDef& fused_graph_def) {
295 TF_ASSERT_OK(session->Create(fused_graph_def));
522 GraphDef fused_graph_def = GraphTransferUtils::BuildFusedGraphDef( local
526 RunFusedGraph(fused_graph_def);
533 GraphDef fused_graph_def; local
535 ReadBinaryProto(Env::Default(), FUSED_MODEL_FILENAME, &fused_graph_def);
536 RunFusedGraph(fused_graph_def);
|
| graph_transferer_test.cc | 414 GraphDef fused_graph_def = GraphTransferUtils::BuildFusedGraphDef( local
418 EXPECT_EQ(3, fused_graph_def.node_size());
|
| hexagon_control_wrapper.cc | 399 const std::vector<string>& outputs, GraphDef* fused_graph_def) {
408 /*require_shape_type=*/true, fused_graph_def));
|
