Searched refs:folded_conv (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/contrib/quantize/python/ |
| D | fold_batch_norms_test.py | 149 folded_conv = g.get_operation_by_name(conv_name + '/Conv2D_Fold')
150 self.assertEqual(folded_conv.type, 'Conv2D')
151 self._AssertInputOpsAre(folded_conv,
153 self._AssertOutputGoesToOps(folded_conv, g, [conv_name + '/post_conv_mul'])
241 folded_conv = g.get_operation_by_name(scope + '/Conv2D_Fold')
242 self.assertEqual(folded_conv.type, 'Conv2D')
244 self._AssertInputOpsAre(folded_conv,
246 self._AssertOutputGoesToOps(folded_conv, g, [scope + '/post_conv_mul'])
320 folded_conv = g.get_operation_by_name(scope + '/Conv2D_Fold')
321 self.assertEqual(folded_conv.type, 'Conv2D')
[all …]
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | arithmetic_optimizer_test.cc | 1562 const NodeDef* folded_conv = node_map.GetNode(conv.node()->name()); in TEST_F() local
1563 ASSERT_NE(folded_conv, nullptr); in TEST_F()
1565 const NodeDef* folded_conv_weights = node_map.GetNode(folded_conv->input(1)); in TEST_F()
1570 const NodeDef* transpose = node_map.GetNode(NodeName(folded_conv->input(0))); in TEST_F()
1640 const NodeDef* folded_conv = node_map.GetNode(conv.node()->name()); in TEST_F() local
1641 CHECK_EQ(inputs.node()->name(), NodeName(folded_conv->input(0))); in TEST_F()
1642 CHECK_EQ(node_map.GetNode(NodeName(folded_conv->input(1)))->op(), "Mul"); in TEST_F()
|