Searched refs:folded_conv (Results 1 - 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/contrib/quantize/python/
H A Dfold_batch_norms_test.py116 folded_conv = g.get_operation_by_name(scope + '/Conv2D_Fold')
117 self.assertEqual(folded_conv.type, 'Conv2D')
118 self._AssertInputOpsAre(folded_conv,
120 self._AssertOutputGoesToOps(folded_conv, g, [scope + '/post_conv_mul'])
185 folded_conv = g.get_operation_by_name(scope + '/Conv2D_Fold')
186 self.assertEqual(folded_conv.type, 'Conv2D')
187 self._AssertInputOpsAre(folded_conv, [scope + '/mul_fold', inputs.op.name])
188 self._AssertOutputGoesToOps(folded_conv, g, [scope + '/post_conv_mul'])
248 folded_conv = g.get_operation_by_name(scope + '/MatMul_Fold')
249 self.assertEqual(folded_conv
[all...]
/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Darithmetic_optimizer_test.cc928 const NodeDef* folded_conv = node_map.GetNode(conv.node()->name()); local
929 CHECK_EQ(node_map.GetNode(NodeName(folded_conv->input(1)))->op(), "Mul");
931 const NodeDef* transpose = node_map.GetNode(NodeName(folded_conv->input(0)));
1000 const NodeDef* folded_conv = node_map.GetNode(conv.node()->name()); local
1001 CHECK_EQ(inputs.node()->name(), NodeName(folded_conv->input(0)));
1002 CHECK_EQ(node_map.GetNode(NodeName(folded_conv->input(1)))->op(), "Mul");

Completed in 229 milliseconds