Home
last modified time | relevance | path

Searched refs:feature_to_gradient_dict (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/python/tpu/
Dtpu_embedding_gradient.py48 feature_to_gradient_dict = collections.OrderedDict(
50 return feature_to_gradient_dict
143 feature_to_gradient_dict = collections.OrderedDict()
166 feature_to_gradient_dict[feature] = gradient
173 feature_to_gradient_dict[feature] = array_ops.zeros(
176 feature_to_gradient_dict[feature] = array_ops.zeros(
179 return feature_to_gradient_dict
Dtpu_embedding.py1880 def generate_send_gradients_op(self, feature_to_gradient_dict, step=None): argument
1906 gradient = feature_to_gradient_dict[feature]