Searched refs:learning_rates (Results 1 – 14 of 14) sorted by relevance
| /external/tensorflow/tensorflow/python/tpu/ops/ |
| D | tpu_ops.py | 250 learning_rates=None, argument
274 if learning_rates is None:
275 learning_rates = []
277 inputs=inputs, learning_rates=learning_rates, config=config, name=name)
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | tpu_embedding_ops.cc | 108 std::vector<shape_inference::ShapeHandle> learning_rates; in __anon355f22aa0302() local
109 TF_RETURN_IF_ERROR(c->input("learning_rates", &learning_rates)); in __anon355f22aa0302()
114 c->WithRank(learning_rates[i], 0, &learning_rates_shape)); in __anon355f22aa0302()
|
| D | ops.pbtxt | 45531 name: "learning_rates"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | SendTPUEmbeddingGradients.pbtxt | 9 name: "learning_rates"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | SendTPUEmbeddingGradients.pbtxt | 9 name: "learning_rates"
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_SendTPUEmbeddingGradients.pbtxt | 16 name: "learning_rates"
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_embedding_v2_test.py | 1201 learning_rates = [starting_lr - (starting_lr - ending_lr) / num_steps * j
1203 cumsum = [sum(learning_rates[0:j]) for j in range(num_steps)]
|
| D | tpu_embedding_v2.py | 664 learning_rates=[math_ops.cast(fn(), dtype=dtypes.float32)
|
| D | tpu_embedding.py | 1918 learning_rates=[
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops.td | 935 learning_rates: A TensorList of float32 scalars, one for each dynamic learning
946 Variadic<TF_Tensor>:$learning_rates,
967 learning_rates: A TensorList of learning rates used for updating the embedding
981 Variadic<TF_Tensor>:$learning_rates,
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 4037 …argspec: "args=[\'inputs\', \'learning_rates\', \'config\', \'name\'], varargs=None, keywords=None…
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 4037 …argspec: "args=[\'inputs\', \'learning_rates\', \'config\', \'name\'], varargs=None, keywords=None…
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 31498 func SendTPUEmbeddingGradients(scope *Scope, inputs []tf.Output, learning_rates []tf.Output, config…
31506 tf.OutputList(inputs), tf.OutputList(learning_rates),
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v2.pbtxt | 68123 name: "learning_rates"
|