| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_embedding_for_serving.py | 28 from tensorflow.python.ops import variables as tf_variables unknown
120 self) -> Dict[tpu_embedding_v2_utils.TableConfig, tf_variables.Variable]:
140 self) -> Dict[Text, Dict[Text, tf_variables.Variable]]:
176 def _ragged_embedding_lookup_with_reduce(table: tf_variables.Variable,
210 tables: Dict[tpu_embedding_v2_utils.TableConfig, tf_variables.Variable],
324 weight: Optional[sparse_tensor.SparseTensor], table: tf_variables.Variable,
366 weight: Optional[ragged_tensor.RaggedTensor], table: tf_variables.Variable,
|
| D | tpu_embedding_base.py | 21 from tensorflow.python.ops import variables as tf_variables unknown
82 trainable: bool) -> Dict[Text, tf_variables.Variable]:
92 return tf_variables.Variable(
|
| D | tpu_embedding_v1.py | 28 from tensorflow.python.ops import variables as tf_variables unknown
144 self) -> Dict[tpu_embedding_v2_utils.TableConfig, tf_variables.Variable]:
154 self) -> Dict[Text, Dict[Text, tf_variables.Variable]]:
309 table: tf_variables.Variable,
354 table: tf_variables.Variable,
|
| D | tpu_embedding_v2_utils.py | 30 from tensorflow.python.ops import variables as tf_variables unknown
37 tf_variables.Variable)
141 tf_variables.Variable]
142 ) -> Dict[Text, tf_variables.Variable]:
|
| D | tpu_embedding_v2.py | 40 from tensorflow.python.ops import variables as tf_variables unknown
505 ) -> Dict[tpu_embedding_v2_utils.TableConfig, tf_variables.Variable]:
796 ) -> Dict[Text, Dict[Text, tf_variables.Variable]]:
817 return tf_variables.Variable(
1488 variables: Dict[Text, Dict[Text, tf_variables.Variable]],
1533 variables: Dict[Text, Dict[Text, tf_variables.Variable]],
1664 next_creator: Callable[..., tf_variables.Variable], *args, **kwargs):
|
| /external/tensorflow/tensorflow/dtensor/python/ |
| D | save_restore.py | 30 from tensorflow.python.ops import variables as tf_variables unknown
88 tensors: List[Union[ops.Tensor, tf_variables.Variable]],
136 def enable_save_as_bf16(variables: List[tf_variables.Variable]):
156 name_tensor_dict: Dict[str, Union[ops.Tensor, tf_variables.Variable]],
229 tf_variables.Variable]]):
|
| /external/tensorflow/tensorflow/python/tpu/tests/ |
| D | tpu_embedding_v2_optimizer_test.py | 31 from tensorflow.python.ops import variables as tf_variables unknown
60 step_counter = tf_variables.Variable(0.0, dtypes.float32)
137 slots[slot] = tf_variables.Variable(
180 slots[slot] = tf_variables.Variable(
|
| /external/tensorflow/tensorflow/python/keras/legacy_tf_layers/ |
| D | base.py | 30 from tensorflow.python.ops import variables as tf_variables unknown
434 existing_variables = set(tf_variables.global_variables())
438 existing_variables = set(tf_variables.global_variables())
486 trainable_variables = tf_variables.trainable_variables()
|
| /external/tensorflow/tensorflow/python/keras/optimizer_v2/ |
| D | optimizer_v2.py | 48 from tensorflow.python.ops import variables as tf_variables unknown
914 weight = tf_variables.Variable(
969 (ops.Tensor, tf_variables.Variable)) or callable(value):
980 aggregation=tf_variables.VariableAggregation.ONLY_FIRST_REPLICA)
993 aggregation=tf_variables.VariableAggregation.ONLY_FIRST_REPLICA)
1166 synchronization=tf_variables.VariableSynchronization.AUTO,
1167 aggregation=tf_variables.VariableAggregation.NONE):
1174 if synchronization == tf_variables.VariableSynchronization.ON_READ:
|
| D | nadam.py | 26 from tensorflow.python.ops import variables as tf_variables unknown
96 aggregation=tf_variables.VariableAggregation.ONLY_FIRST_REPLICA)
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | base_layer_utils.py | 33 from tensorflow.python.ops import variables as tf_variables unknown
61 synchronization=tf_variables.VariableSynchronization.AUTO,
62 aggregation=tf_variables.VariableAggregation.NONE,
127 return tf_variables.VariableV1(
|
| D | base_layer_v1.py | 59 from tensorflow.python.ops import variables as tf_variables unknown
327 synchronization=tf_variables.VariableSynchronization.AUTO,
328 aggregation=tf_variables.VariableAggregation.NONE,
395 if synchronization == tf_variables.VariableSynchronization.ON_READ:
2189 if isinstance(existing_value, tf_variables.Variable):
2250 if not isinstance(val, tf_variables.Variable):
|
| D | base_layer.py | 69 from tensorflow.python.ops import variables as tf_variables unknown
543 synchronization=tf_variables.VariableSynchronization.AUTO,
544 aggregation=tf_variables.VariableAggregation.NONE,
605 if synchronization == tf_variables.VariableSynchronization.ON_READ:
2758 if isinstance(existing_value, tf_variables.Variable):
2817 if not isinstance(val, tf_variables.Variable):
|
| /external/tensorflow/tensorflow/python/keras/layers/legacy_rnn/ |
| D | rnn_cell_impl.py | 49 from tensorflow.python.ops import variables as tf_variables unknown
254 variable in tf_variables.trainable_variables() or
256 list(variable)[0] in tf_variables.trainable_variables()))
|