Searched refs:stop_gradient (Results 1 – 25 of 34) sorted by relevance
12
| /external/tensorflow/tensorflow/python/ops/ |
| D | gradients_impl.py | 367 math_ops.multiply(grad_elem, array_ops.stop_gradient(v_elem))
|
| D | gradients_test.py | 360 if "a" in stop_gradients: a = array_ops.stop_gradient(a)
362 if "b" in stop_gradients: b = array_ops.stop_gradient(b)
364 if "c" in stop_gradients: c = array_ops.stop_gradient(c)
366 if "d" in stop_gradients: d = array_ops.stop_gradient(d)
480 e = array_ops.stop_gradient(c) + d
696 out = array_ops.stop_gradient(inp)
|
| D | nn_grad.py | 231 return (array_ops.stop_gradient(op.inputs[0]),
673 return (array_ops.stop_gradient(op.inputs[0]),
994 math_ops.squared_difference(x, array_ops.stop_gradient(mean_x)),
|
| D | nn_impl.py | 1356 math_ops.squared_difference(y, array_ops.stop_gradient(mean)),
1895 array_ops.stop_gradient(s) for s in sampled_values)
2401 labels = array_ops.stop_gradient(labels, name="labels_stop_gradient")
|
| D | while_v2.py | 296 output_loop_vars = [array_ops.stop_gradient(t) for t in output_loop_vars]
|
| /external/tensorflow/tensorflow/python/kernel_tests/linalg/sparse/ |
| D | csr_sparse_matrix_dense_mat_mul_grad_test.py | 38 locs = array_ops.stop_gradient(array_ops.where(math_ops.abs(dense_t) > 0))
|
| D | csr_sparse_matrix_grad_test.py | 37 locs = array_ops.stop_gradient(array_ops.where(math_ops.abs(dense_t) > 0))
|
| D | csr_sparse_matrix_sparse_mat_mul_grad_test.py | 38 locs = array_ops.stop_gradient(array_ops.where(math_ops.abs(dense_t) > 0))
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | normalization_v2.py | 177 math_ops.squared_difference(y, array_ops.stop_gradient(mean)),
|
| D | normalization.py | 861 r = _broadcast(array_ops.stop_gradient(r, name='renorm_r'))
862 d = _broadcast(array_ops.stop_gradient(d, name='renorm_d'))
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_StopGradient.pbtxt | 47 z = x - tf.stop_gradient(tf.reduce_max(x))
|
| /external/tensorflow/tensorflow/python/kernel_tests/random/ |
| D | random_grad_test.py | 183 sample_sg = array_ops.stop_gradient(sample)
|
| /external/tensorflow/tensorflow/python/ops/linalg/sparse/ |
| D | sparse_csr_matrix_grad.py | 42 grad, array_ops.stop_gradient(array_ops.where(math_ops.abs(grad) > 0)))
|
| D | sparse_csr_matrix_ops.py | 345 indices = array_ops.stop_gradient(array_ops.where(value))
|
| /external/tensorflow/tensorflow/python/kernel_tests/distributions/ |
| D | util_test.py | 718 zeros_like_x_pl = (x_pl * array_ops.stop_gradient(x_pl - 1.)
719 - array_ops.stop_gradient(x_pl * (x_pl - 1.)))
793 zeros_like_x_pl = (x_pl * array_ops.stop_gradient(x_pl - 1.)
794 - array_ops.stop_gradient(x_pl * (x_pl - 1.)))
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | control_flow_ops_py_test.py | 3976 r = gradients_impl.gradients(array_ops.stop_gradient(rx), y)[0]
3978 r = gradients_impl.gradients(array_ops.stop_gradient(ry), y)[0]
3982 array_ops.stop_gradient(math_ops.square(rx)), y)[0]
3985 array_ops.stop_gradient(math_ops.add(rx, ry)), x)[0]
3988 array_ops.stop_gradient(math_ops.add(rx, ry)), y)[0]
3994 math_ops.add(rx, array_ops.stop_gradient(ry)), y)[0]
3997 math_ops.add(array_ops.stop_gradient(rx), ry), y)[0]
4009 y1 = array_ops.stop_gradient(math_ops.square(y))
4029 y1 = array_ops.stop_gradient(math_ops.square(y, name="stopped"))
4056 grad_theta_stopped = array_ops.stop_gradient(grad_theta)
[all …]
|
| /external/tensorflow/tensorflow/python/ops/losses/ |
| D | losses_impl.py | 785 onehot_labels = array_ops.stop_gradient(
|
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | array_test.py | 340 for op in [array_ops.identity, array_ops.stop_gradient]:
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | backprop_test.py | 608 lambda x: array_ops.stop_gradient(math_ops.argmax(x)))
616 return array_ops.stop_gradient(i)
1019 return array_ops.stop_gradient(z)
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.keras.backend.pbtxt | 520 name: "stop_gradient"
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.keras.backend.pbtxt | 528 name: "stop_gradient"
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | backend.py | 4223 def stop_gradient(variables): function
4236 return map(array_ops.stop_gradient, variables)
4237 return array_ops.stop_gradient(variables)
|
| D | backend_test.py | 256 y = backend.stop_gradient(x)
261 ys = backend.stop_gradient(xs)
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/tests/ |
| D | prepare-tf.mlir | 431 func @stop_gradient(%arg0: tensor<3xi32>) -> tensor<3xi32> {
435 // CHECK-LABEL: stop_gradient
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | tensor_util_test.py | 1046 tf_val = array_ops.stop_gradient(input_)
|
12