Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/python/ops/
Dgradients_test.py41 from tensorflow.python.ops import custom_gradient
961 dependent_vars = custom_gradient._get_dependent_variables(
977 dependent_vars = custom_gradient._get_dependent_variables(
992 dependent_vars = custom_gradient._get_dependent_variables(
1006 dependent_vars = custom_gradient._get_dependent_variables(
1023 dependent_vars = custom_gradient._get_dependent_variables(
1042 dependent_vars = custom_gradient._get_dependent_variables(
1051 @custom_gradient.custom_gradient
1070 dependent_vars = custom_gradient._get_dependent_variables(
1079 @custom_gradient.custom_gradient
[all …]
Dgradients.py25 from tensorflow.python.ops.custom_gradient import custom_gradient
Dcustom_gradient.py50 def custom_gradient(f=None): function
207 return lambda f: custom_gradient(f=f)
511 @custom_gradient
521 @custom_gradient
602 @custom_gradient
Dgradient_checker_v2_test.py29 from tensorflow.python.ops import custom_gradient
220 @custom_gradient.custom_gradient
242 @custom_gradient.custom_gradient
Dctc_ops.py35 from tensorflow.python.ops import custom_gradient
1039 @custom_gradient.custom_gradient
Dnn_impl.py30 from tensorflow.python.ops import custom_gradient
514 @custom_gradient.custom_gradient
Dcond_v2.py42 from tensorflow.python.ops import custom_gradient
335 custom_gradient.copy_handle_data(external_t, internal_t)
Dwhile_v2.py44 from tensorflow.python.ops import custom_gradient
1308 custom_gradient.copy_handle_data(src_t, tgt_t)
Dcontrol_flow_ops_test.py46 from tensorflow.python.ops import custom_gradient
1536 @custom_gradient.custom_gradient
/external/tensorflow/tensorflow/python/keras/mixed_precision/
Dtest_util.py27 from tensorflow.python.ops import custom_gradient
49 @custom_gradient.custom_gradient
92 @custom_gradient.custom_gradient
/external/tensorflow/tensorflow/python/eager/
Dtape_test.py29 from tensorflow.python.ops import custom_gradient
38 @custom_gradient.custom_gradient
54 @custom_gradient.custom_gradient
Dbackprop_test.py43 from tensorflow.python.ops import custom_gradient
198 @custom_gradient.custom_gradient
239 @custom_gradient.custom_gradient
253 @custom_gradient.custom_gradient
1229 @custom_gradient.custom_gradient
1267 @custom_gradient.custom_gradient
1295 @custom_gradient.custom_gradient
1323 @custom_gradient.custom_gradient
1440 @custom_gradient.custom_gradient
1457 @custom_gradient.custom_gradient
[all …]
Dforwardprop_test.py40 from tensorflow.python.ops import custom_gradient
429 @custom_gradient.custom_gradient
443 @custom_gradient.recompute_grad
454 @custom_gradient.custom_gradient
507 @custom_gradient.custom_gradient
Dfunction.py60 from tensorflow.python.ops import custom_gradient
610 custom_gradient.copy_handle_data(func_graph_output, outputs[i])
756 custom_gradient.copy_handle_data(func_graph_output, op.outputs[i])
923 custom_gradient.copy_handle_data(output, gradient_placeholder)
1024 custom_gradient.copy_handle_data(capture, input_placeholder)
2189 custom_gradient.copy_handle_data(self.outputs[j], result[j])
/external/tensorflow/tensorflow/python/training/experimental/
Dloss_scale_optimizer_test.py32 from tensorflow.python.ops import custom_gradient
86 @custom_gradient.custom_gradient
/external/tensorflow/tensorflow/python/debug/lib/
Dcheck_numerics_callback_test.py35 from tensorflow.python.ops import custom_gradient
343 @custom_gradient.custom_gradient
/external/tensorflow/tensorflow/python/distribute/
Ddistribute_lib.py218 from tensorflow.python.ops import custom_gradient
3128 @custom_gradient.custom_gradient
3293 @custom_gradient.custom_gradient
/external/tensorflow/tensorflow/python/kernel_tests/
Dwhile_v2_test.py44 from tensorflow.python.ops import custom_gradient
114 @custom_gradient.custom_gradient
1295 @custom_gradient.custom_gradient
1335 @custom_gradient.custom_gradient
Dresource_variable_ops_test.py44 from tensorflow.python.ops import custom_gradient
329 custom_gradient.copy_handle_data(v.handle, identity[1])
823 @custom_gradient.custom_gradient
/external/tensorflow/tensorflow/python/saved_model/
Dload.py37 from tensorflow.python.ops import custom_gradient
368 custom_gradient.copy_handle_data(handle, internal_capture)
370 custom_gradient.copy_handle_data(bound_input, internal_capture)
/external/tensorflow/tensorflow/python/framework/
Dfunc_graph.py42 from tensorflow.python.ops import custom_gradient
1144 custom_gradient.copy_handle_data(value, placeholder)
1231 custom_gradient.copy_handle_data(arg, placeholder)
/external/tensorflow/tensorflow/python/ops/parallel_for/
Dpfor.py46 from tensorflow.python.ops import custom_gradient
1579 custom_gradient.copy_handle_data(old_output, new_output)
3637 custom_gradient.copy_handle_data(original_tensor, result)
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.pbtxt592 name: "custom_gradient"
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.pbtxt1080 name: "custom_gradient"
/external/tensorflow/tensorflow/python/
DBUILD3429 "ops/custom_gradient.py",

12