Searched refs:_get_tensor (Results 1 – 2 of 2) sorted by relevance
169 def _get_tensor(name): function174 return _get_tensor(tensor_info.name)177 _get_tensor(tensor_info.coo_sparse.indices_tensor_name),178 _get_tensor(tensor_info.coo_sparse.values_tensor_name),179 _get_tensor(tensor_info.coo_sparse.dense_shape_tensor_name))185 components = [_get_tensor(component.name) for component in
108 def _get_tensor(self, is_finite): member in DynamicLossScaleTest138 grad = self._get_tensor(is_finite)