Home
last modified time | relevance | path

Searched refs:aggregation_method (Results 1 – 23 of 23) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Dgradients_impl.py46 aggregation_method=None, argument
167 gate_gradients, aggregation_method, stop_gradients,
178 aggregation_method=None, argument
313 aggregation_method, stop_gradients,
378 aggregation_method=None): argument
408 "aggregation_method": aggregation_method
443 aggregation_method=None, argument
475 aggregation_method=aggregation_method)
H A Dgradients_util.py474 aggregation_method=None, argument
500 aggregation_method, stop_gradients,
597 aggregation_method)
941 aggregation_method=None): argument
966 if aggregation_method is None:
967 aggregation_method = AggregationMethod.DEFAULT
971 if aggregation_method not in valid_aggregation_methods:
995 if aggregation_method in [
H A Dgradients_test.py184 aggregation_method=gradients.AggregationMethod.
197 z, [x, y], aggregation_method=gradients.AggregationMethod.ADD_N)
210 aggregation_method=gradients.AggregationMethod.EXPERIMENTAL_TREE)
/aosp_15_r20/external/tensorflow/tensorflow/python/training/experimental/
H A Dloss_scale_optimizer.py84 aggregation_method=None, argument
119 aggregation_method=aggregation_method,
/aosp_15_r20/external/tensorflow/tensorflow/python/training/
H A Doptimizer.py431 gate_gradients=GATE_OP, aggregation_method=None, argument
476 aggregation_method=aggregation_method,
492 aggregation_method=None, argument
603 aggregation_method=aggregation_method,
H A Doptimizer_test.py78 aggregation_method=gradients_util.AggregationMethod.
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v1/
H A Dtensorflow.train.-optimizer.pbtxt28 …argspec: "args=[\'self\', \'loss\', \'var_list\', \'gate_gradients\', \'aggregation_method\', \'co…
44 …\', \'loss\', \'global_step\', \'var_list\', \'gate_gradients\', \'aggregation_method\', \'colocat…
H A Dtensorflow.train.-gradient-descent-optimizer.pbtxt29 …argspec: "args=[\'self\', \'loss\', \'var_list\', \'gate_gradients\', \'aggregation_method\', \'co…
45 …\', \'loss\', \'global_step\', \'var_list\', \'gate_gradients\', \'aggregation_method\', \'colocat…
H A Dtensorflow.train.-adagrad-d-a-optimizer.pbtxt29 …argspec: "args=[\'self\', \'loss\', \'var_list\', \'gate_gradients\', \'aggregation_method\', \'co…
45 …\', \'loss\', \'global_step\', \'var_list\', \'gate_gradients\', \'aggregation_method\', \'colocat…
H A Dtensorflow.train.-momentum-optimizer.pbtxt29 …argspec: "args=[\'self\', \'loss\', \'var_list\', \'gate_gradients\', \'aggregation_method\', \'co…
45 …\', \'loss\', \'global_step\', \'var_list\', \'gate_gradients\', \'aggregation_method\', \'colocat…
H A Dtensorflow.train.-adagrad-optimizer.pbtxt29 …argspec: "args=[\'self\', \'loss\', \'var_list\', \'gate_gradients\', \'aggregation_method\', \'co…
45 …\', \'loss\', \'global_step\', \'var_list\', \'gate_gradients\', \'aggregation_method\', \'colocat…
H A Dtensorflow.train.-adadelta-optimizer.pbtxt29 …argspec: "args=[\'self\', \'loss\', \'var_list\', \'gate_gradients\', \'aggregation_method\', \'co…
45 …\', \'loss\', \'global_step\', \'var_list\', \'gate_gradients\', \'aggregation_method\', \'colocat…
H A Dtensorflow.mixed_precision.-mixed-precision-loss-scale-optimizer.pbtxt29 …argspec: "args=[\'self\', \'loss\', \'var_list\', \'gate_gradients\', \'aggregation_method\', \'co…
45 …\', \'loss\', \'global_step\', \'var_list\', \'gate_gradients\', \'aggregation_method\', \'colocat…
H A Dtensorflow.train.-proximal-adagrad-optimizer.pbtxt29 …argspec: "args=[\'self\', \'loss\', \'var_list\', \'gate_gradients\', \'aggregation_method\', \'co…
45 …\', \'loss\', \'global_step\', \'var_list\', \'gate_gradients\', \'aggregation_method\', \'colocat…
H A Dtensorflow.train.-proximal-gradient-descent-optimizer.pbtxt29 …argspec: "args=[\'self\', \'loss\', \'var_list\', \'gate_gradients\', \'aggregation_method\', \'co…
45 …\', \'loss\', \'global_step\', \'var_list\', \'gate_gradients\', \'aggregation_method\', \'colocat…
H A Dtensorflow.train.-r-m-s-prop-optimizer.pbtxt29 …argspec: "args=[\'self\', \'loss\', \'var_list\', \'gate_gradients\', \'aggregation_method\', \'co…
45 …\', \'loss\', \'global_step\', \'var_list\', \'gate_gradients\', \'aggregation_method\', \'colocat…
H A Dtensorflow.train.-adam-optimizer.pbtxt29 …argspec: "args=[\'self\', \'loss\', \'var_list\', \'gate_gradients\', \'aggregation_method\', \'co…
45 …\', \'loss\', \'global_step\', \'var_list\', \'gate_gradients\', \'aggregation_method\', \'colocat…
H A Dtensorflow.train.experimental.-mixed-precision-loss-scale-optimizer.pbtxt29 …argspec: "args=[\'self\', \'loss\', \'var_list\', \'gate_gradients\', \'aggregation_method\', \'co…
45 …\', \'loss\', \'global_step\', \'var_list\', \'gate_gradients\', \'aggregation_method\', \'colocat…
H A Dtensorflow.train.-ftrl-optimizer.pbtxt29 …argspec: "args=[\'self\', \'loss\', \'var_list\', \'gate_gradients\', \'aggregation_method\', \'co…
45 …\', \'loss\', \'global_step\', \'var_list\', \'gate_gradients\', \'aggregation_method\', \'colocat…
H A Dtensorflow.tpu.-cross-shard-optimizer.pbtxt45 …\', \'loss\', \'global_step\', \'var_list\', \'gate_gradients\', \'aggregation_method\', \'colocat…
H A Dtensorflow.train.-sync-replicas-optimizer.pbtxt57 …\', \'loss\', \'global_step\', \'var_list\', \'gate_gradients\', \'aggregation_method\', \'colocat…
H A Dtensorflow.pbtxt1429 …, \'name\', \'colocate_gradients_with_ops\', \'gate_gradients\', \'aggregation_method\', \'stop_gr…
1449 …, \'name\', \'colocate_gradients_with_ops\', \'gate_gradients\', \'aggregation_method\'], varargs=…
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v2/
H A Dtensorflow.pbtxt709 …argspec: "args=[\'ys\', \'xs\', \'grad_ys\', \'name\', \'gate_gradients\', \'aggregation_method\',…
729 …argspec: "args=[\'ys\', \'xs\', \'gate_gradients\', \'aggregation_method\', \'name\'], varargs=Non…