Searched defs:op_fn (Results 1 – 3 of 3) sorted by relevance
163 def testKerasLayers(self, value, op_fn, atol=1e-6): argument191 def testBatchNorm(self, value, op_fn): argument208 def testBatchNormLayerParamGrads(self, value, op_fn): argument
82 void op_fn(int i) { in op_fn() function
472 def testElementwiseNNOps(self, value, op_fn): argument