Home
last modified time | relevance | path

Searched refs:custom_gradient (Results 1 – 19 of 19) sorted by relevance

/external/tensorflow/tensorflow/python/keras/mixed_precision/experimental/
Dtest_util.py23 from tensorflow.python.ops import custom_gradient
43 @custom_gradient.custom_gradient
/external/tensorflow/tensorflow/python/eager/
Dtape_test.py28 from tensorflow.python.ops import custom_gradient
36 @custom_gradient.custom_gradient
52 @custom_gradient.custom_gradient
Dbackprop_test.py36 from tensorflow.python.ops import custom_gradient
137 @custom_gradient.custom_gradient
176 @custom_gradient.custom_gradient
188 @custom_gradient.custom_gradient
1080 @custom_gradient.custom_gradient
1106 @custom_gradient.custom_gradient
1133 @custom_gradient.custom_gradient
1248 @custom_gradient.custom_gradient
1352 @custom_gradient.custom_gradient
1455 @custom_gradient.custom_gradient
Dfunction.py47 from tensorflow.python.ops import custom_gradient
439 custom_gradient.copy_handle_data(func_graph_output, outputs[i])
651 custom_gradient.copy_handle_data(func_graph_output, op.outputs[i])
/external/tensorflow/tensorflow/python/ops/
Dgradients_test.py42 from tensorflow.python.ops import custom_gradient
866 @custom_gradient.custom_gradient
883 @custom_gradient.custom_gradient
903 @custom_gradient.custom_gradient
919 @custom_gradient.custom_gradient
948 @custom_gradient.custom_gradient
983 @custom_gradient.custom_gradient
987 @custom_gradient.custom_gradient
1007 @custom_gradient.custom_gradient
1037 @custom_gradient.custom_gradient
[all …]
Dgradients.py24 from tensorflow.python.ops.custom_gradient import custom_gradient
Dgradient_checker_v2_test.py29 from tensorflow.python.ops import custom_gradient
147 @custom_gradient.custom_gradient
164 @custom_gradient.custom_gradient
Dcustom_gradient.py80 def custom_gradient(f): function
Dwhile_v2.py37 from tensorflow.python.ops import custom_gradient
932 custom_gradient.copy_handle_data(src_t, tgt_t)
/external/tensorflow/tensorflow/contrib/eager/python/
Dtfe.py135 from tensorflow.python.ops.custom_gradient import custom_gradient
Dtfe_test.py70 @tfe.custom_gradient
/external/tensorflow/tensorflow/contrib/layers/python/layers/
Drev_block_lib.py44 from tensorflow.python.ops import custom_gradient
353 @custom_gradient.custom_gradient
589 @custom_gradient.custom_gradient
/external/tensorflow/tensorflow/python/distribute/
Ddistribute_lib.py37 from tensorflow.python.ops import custom_gradient
1507 @custom_gradient.custom_gradient
/external/tensorflow/tensorflow/python/framework/
Dfunc_graph.py35 from tensorflow.python.ops import custom_gradient
838 custom_gradient.copy_handle_data(value, placeholder)
/external/tensorflow/tensorflow/python/kernel_tests/
Dresource_variable_ops_test.py41 from tensorflow.python.ops import custom_gradient
606 @custom_gradient.custom_gradient
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.pbtxt544 name: "custom_gradient"
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.pbtxt1044 name: "custom_gradient"
/external/tensorflow/tensorflow/python/
DBUILD2459 "ops/custom_gradient.py",
/external/tensorflow/
DRELEASE.md541 * Easily customize gradient computation with `tf.custom_gradient`.