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