Searched refs:learning_rates (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/python/tpu/ops/ |
| D | tpu_ops.py | 247 learning_rates=None, argument
271 if learning_rates is None:
272 learning_rates = []
274 inputs=inputs, learning_rates=learning_rates, config=config, name=name)
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | tpu_embedding_ops.cc | 375 std::vector<shape_inference::ShapeHandle> learning_rates; in __anon8eedc5730b02() local
376 TF_RETURN_IF_ERROR(c->input("learning_rates", &learning_rates)); in __anon8eedc5730b02()
381 c->WithRank(learning_rates[i], 0, &learning_rates_shape)); in __anon8eedc5730b02()
|
| D | ops.pbtxt | 34113 name: "learning_rates"
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_SendTPUEmbeddingGradients.pbtxt | 16 name: "learning_rates"
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 3281 …argspec: "args=[\'inputs\', \'learning_rates\', \'config\', \'name\'], varargs=None, keywords=None…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 3281 …argspec: "args=[\'inputs\', \'learning_rates\', \'config\', \'name\'], varargs=None, keywords=None…
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 27158 func SendTPUEmbeddingGradients(scope *Scope, inputs []tf.Output, learning_rates []tf.Output, config…
27166 tf.OutputList(inputs), tf.OutputList(learning_rates),
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v2.pbtxt | 68786 name: "learning_rates"
|
| D | ops_history.v1.pbtxt | 69413 name: "learning_rates"
|