Searched refs:compute_op (Results 1 – 2 of 2) sorted by relevance
22 def _assertOpAndComposite(self, vars_, compute_op, compute_composite, kwargs, argument31 y = compute_op(**op_kwargs) # uses op and decomposites by the graph pass.
3774 def _wrap_2d_function(inputs, compute_op, dim=-1, name=None): argument3816 return compute_op(inputs, name=name)3853 outputs = compute_op(inputs)