Home
last modified time | relevance | path

Searched refs:tf_variables (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/python/tpu/
Dtpu_embedding_v2_utils.py35 from tensorflow.python.ops import variables as tf_variables unknown
42 tf_variables.Variable)
145 tf_variables.Variable]
146 ) -> Dict[Text, tf_variables.Variable]:
Dtpu_embedding_v2.py46 from tensorflow.python.ops import variables as tf_variables unknown
414 ) -> Dict[tpu_embedding_v2_utils.TableConfig, tf_variables.Variable]:
787 ) -> Dict[Text, Dict[Text, tf_variables.Variable]]:
808 return tf_variables.Variable(
1369 variables: Dict[Text, Dict[Text, tf_variables.Variable]],
1414 variables: Dict[Text, Dict[Text, tf_variables.Variable]],
1477 table: tf_variables.Variable,
1711 next_creator: Callable[..., tf_variables.Variable], *args, **kwargs):
Dtpu_embedding_v2_test.py48 from tensorflow.python.ops import variables as tf_variables unknown
1137 step_counter = tf_variables.Variable(0.0, dtypes.float32)
1213 slots[slot] = tf_variables.Variable(
1256 slots[slot] = tf_variables.Variable(
/external/tensorflow/tensorflow/python/keras/optimizer_v2/
Doptimizer_v2.py53 from tensorflow.python.ops import variables as tf_variables unknown
908 weight = tf_variables.Variable(
963 (ops.Tensor, tf_variables.Variable)) or callable(value):
974 aggregation=tf_variables.VariableAggregation.ONLY_FIRST_REPLICA)
987 aggregation=tf_variables.VariableAggregation.ONLY_FIRST_REPLICA)
1162 synchronization=tf_variables.VariableSynchronization.AUTO,
1163 aggregation=tf_variables.VariableAggregation.NONE):
1170 if synchronization == tf_variables.VariableSynchronization.ON_READ:
Dnadam.py29 from tensorflow.python.ops import variables as tf_variables unknown
99 aggregation=tf_variables.VariableAggregation.ONLY_FIRST_REPLICA)
/external/tensorflow/tensorflow/python/keras/legacy_tf_layers/
Dbase.py34 from tensorflow.python.ops import variables as tf_variables unknown
433 existing_variables = set(tf_variables.global_variables())
437 existing_variables = set(tf_variables.global_variables())
479 trainable_variables = tf_variables.trainable_variables()
/external/tensorflow/tensorflow/python/keras/layers/
Dnormalization.py38 from tensorflow.python.ops import variables as tf_variables unknown
451 synchronization=tf_variables.VariableSynchronization.ON_READ,
453 aggregation=tf_variables.VariableAggregation.MEAN,
461 synchronization=tf_variables.VariableSynchronization.ON_READ,
463 aggregation=tf_variables.VariableAggregation.MEAN,
480 synchronization=tf_variables.VariableSynchronization.ON_READ,
482 aggregation=tf_variables.VariableAggregation.MEAN,
501 synchronization=tf_variables.VariableSynchronization.ON_READ,
503 aggregation=tf_variables.VariableAggregation.MEAN,
/external/tensorflow/tensorflow/python/keras/engine/
Dbase_layer_utils.py36 from tensorflow.python.ops import variables as tf_variables unknown
64 synchronization=tf_variables.VariableSynchronization.AUTO,
65 aggregation=tf_variables.VariableAggregation.NONE,
130 return tf_variables.VariableV1(
Dbase_layer_v1.py64 from tensorflow.python.ops import variables as tf_variables unknown
329 synchronization=tf_variables.VariableSynchronization.AUTO,
330 aggregation=tf_variables.VariableAggregation.NONE,
396 if synchronization == tf_variables.VariableSynchronization.ON_READ:
2181 if isinstance(existing_value, tf_variables.Variable):
2242 if not isinstance(val, tf_variables.Variable):
Dbase_layer.py76 from tensorflow.python.ops import variables as tf_variables unknown
533 synchronization=tf_variables.VariableSynchronization.AUTO,
534 aggregation=tf_variables.VariableAggregation.NONE,
595 if synchronization == tf_variables.VariableSynchronization.ON_READ:
2827 if isinstance(existing_value, tf_variables.Variable):
2886 if not isinstance(val, tf_variables.Variable):
/external/tensorflow/tensorflow/python/keras/layers/legacy_rnn/
Drnn_cell_impl.py52 from tensorflow.python.ops import variables as tf_variables unknown
255 variable in tf_variables.trainable_variables() or
257 list(variable)[0] in tf_variables.trainable_variables()))
/external/tensorflow/tensorflow/python/keras/
Dmetrics.py73 from tensorflow.python.ops import variables as tf_variables unknown
291 aggregation=tf_variables.VariableAggregation.SUM,
292 synchronization=tf_variables.VariableSynchronization.ON_READ,
303 synchronization = tf_variables.VariableSynchronization.ON_WRITE