Home
last modified time | relevance | path

Searched refs:_AssertShapesMatch (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/contrib/quantize/python/
Dfold_batch_norms.py791 _AssertShapesMatch('mul_fold', mul_fold.inputs[0], mul_fold.outputs[0])
814 _AssertShapesMatch('add_fold', add_fold.inputs[0], add_fold.outputs[0])
940 def _AssertShapesMatch(op_name, in_tensor, out_tensor): function