Searched refs:_resource_scatter_update (Results 1 – 2 of 2) sorted by relevance
163 m_t = self._resource_scatter_update(m, indices, m_t_slice)171 v_t = self._resource_scatter_update(v, indices, v_t_slice)
1306 def _resource_scatter_update(self, x, i, v): member in OptimizerV2