Searched refs:_try_replicate_spec_for_scalar_tensor (Results 1 – 1 of 1) sorted by relevance
371 self._try_replicate_spec_for_scalar_tensor(op_call, arg, mesh)391 kwargs_schema[k] = self._try_replicate_spec_for_scalar_tensor(443 def _try_replicate_spec_for_scalar_tensor( member in OpDispatcher