| /external/tensorflow/tensorflow/contrib/framework/python/framework/ |
| D | graph_util_test.py | 50 fused_graph_def = graph_util.fuse_op(
52 self.assertEqual(len(fused_graph_def.node), 4)
53 self.assertEqual(fused_graph_def.node[0].name, 'A')
54 self.assertEqual(fused_graph_def.node[1].name, 'FusedOp')
55 self.assertEqual(fused_graph_def.node[1].input[0], 'A')
56 self.assertEqual(fused_graph_def.node[1].op, 'Op2')
57 self.assertEqual(fused_graph_def.node[1].attr['_output_quantized'].b, True)
58 self.assertEqual(fused_graph_def.node[1].attr['_output_types'].list.type,
60 self.assertEqual(fused_graph_def.node[2].name, 'D')
61 self.assertEqual(fused_graph_def.node[3].name, 'E')
[all …]
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | fold_batch_norms_test.cc | 74 GraphDef fused_graph_def; in TestFoldBatchNormsConv2D() local
76 FoldBatchNorms(original_graph_def, {{}, {"output"}}, &fused_graph_def)); in TestFoldBatchNormsConv2D()
79 TF_ASSERT_OK(fused_session->Create(fused_graph_def)); in TestFoldBatchNormsConv2D()
85 for (const NodeDef& node : fused_graph_def.node()) { in TestFoldBatchNormsConv2D()
125 GraphDef fused_graph_def; in TestFoldBatchNormsDepthwiseConv2dNative() local
127 FoldBatchNorms(original_graph_def, {{}, {"output"}}, &fused_graph_def)); in TestFoldBatchNormsDepthwiseConv2dNative()
130 TF_ASSERT_OK(fused_session->Create(fused_graph_def)); in TestFoldBatchNormsDepthwiseConv2dNative()
136 for (const NodeDef& node : fused_graph_def.node()) { in TestFoldBatchNormsDepthwiseConv2dNative()
185 GraphDef fused_graph_def; in TestFoldBatchNormsConv2DShared() local
187 original_graph_def, {{}, {"output", "output_2"}}, &fused_graph_def)); in TestFoldBatchNormsConv2DShared()
[all …]
|
| D | fuse_convolutions_test.cc | 83 GraphDef fused_graph_def; in TestFuseResizePadAndConv() local
85 &fused_graph_def)); in TestFuseResizePadAndConv()
88 TF_ASSERT_OK(fused_session->Create(fused_graph_def)); in TestFuseResizePadAndConv()
94 for (const NodeDef& node : fused_graph_def.node()) { in TestFuseResizePadAndConv()
133 GraphDef fused_graph_def; in TestFuseResizeAndConv() local
135 &fused_graph_def)); in TestFuseResizeAndConv()
138 TF_ASSERT_OK(fused_session->Create(fused_graph_def)); in TestFuseResizeAndConv()
144 for (const NodeDef& node : fused_graph_def.node()) { in TestFuseResizeAndConv()
185 GraphDef fused_graph_def; in TestFusePadAndConv() local
187 FusePadAndConv(original_graph_def, {{}, {"output"}}, &fused_graph_def)); in TestFusePadAndConv()
[all …]
|
| D | fold_old_batch_norms_test.cc | 108 GraphDef fused_graph_def; in TestFoldOldBatchNorms() local
110 &fused_graph_def)); in TestFoldOldBatchNorms()
113 TF_ASSERT_OK(fused_session->Create(fused_graph_def)); in TestFoldOldBatchNorms()
119 for (const NodeDef& node : fused_graph_def.node()) { in TestFoldOldBatchNorms()
186 GraphDef fused_graph_def; in TestFoldOldBatchNormsAfterDepthwiseConv2dNative() local
188 &fused_graph_def)); in TestFoldOldBatchNormsAfterDepthwiseConv2dNative()
191 TF_ASSERT_OK(fused_session->Create(fused_graph_def)); in TestFoldOldBatchNormsAfterDepthwiseConv2dNative()
197 for (const NodeDef& node : fused_graph_def.node()) { in TestFoldOldBatchNormsAfterDepthwiseConv2dNative()
263 GraphDef fused_graph_def; in TestFoldFusedBatchNorms() local
265 &fused_graph_def)); in TestFoldFusedBatchNorms()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | remote_fused_graph_execute_op_test.cc | 220 GraphDef* fused_graph_def) final { in FuseRemoteGraph() argument
221 *fused_graph_def = original_graph_def; in FuseRemoteGraph()
261 GraphDef fused_graph_def; in RewriteGraphToFusedGraph() local
262 TF_CHECK_OK(root.ToGraphDef(&fused_graph_def)); in RewriteGraphToFusedGraph()
263 *fused_graph = fused_graph_def; in RewriteGraphToFusedGraph()
|
| D | remote_fused_graph_execute_op_test_utils.cc | 118 const std::vector<string>& outputs, GraphDef* fused_graph_def) { in FuseRemoteGraph() argument
122 /*require_shape_type=*/false, fused_graph_def); in FuseRemoteGraph()
|
| D | i_remote_fused_graph_executor.h | 67 GraphDef* fused_graph_def) = 0;
|
| D | remote_fused_graph_execute_op_test_utils.h | 79 GraphDef* fused_graph_def) final;
|
| /external/tensorflow/tensorflow/core/kernels/hexagon/ |
| D | hexagon_graph_execution_test.cc | 283 static void RunFusedGraph(const GraphDef& fused_graph_def) { in RunFusedGraph() argument
297 TF_ASSERT_OK(session->Create(fused_graph_def)); in RunFusedGraph()
524 GraphDef fused_graph_def = GraphTransferUtils::BuildFusedGraphDef( in TEST() local
528 RunFusedGraph(fused_graph_def); in TEST()
535 GraphDef fused_graph_def; in TEST() local
537 ReadBinaryProto(Env::Default(), FUSED_MODEL_FILENAME, &fused_graph_def); in TEST()
538 RunFusedGraph(fused_graph_def); in TEST()
|
| D | hexagon_control_wrapper.h | 54 GraphDef* fused_graph_def) final;
|
| D | hexagon_control_wrapper.cc | 398 const std::vector<string>& outputs, GraphDef* fused_graph_def) { in FuseRemoteGraph() argument
407 /*require_shape_type=*/true, fused_graph_def)); in FuseRemoteGraph()
|
| D | graph_transferer_test.cc | 411 GraphDef fused_graph_def = GraphTransferUtils::BuildFusedGraphDef( in TEST_F() local
415 EXPECT_EQ(3, fused_graph_def.node_size()); in TEST_F()
|