| /external/tensorflow/tensorflow/compiler/tests/ |
| fused_batchnorm_test.py | 164 grad_val = np.random.random_sample(x_shape).astype(np.float32)
182 grad: grad_val,
190 x_val, grad_val, scale_val, mean_val, var_val, epsilon, "NHWC")
202 grad_val = np.random.random_sample(x_shape).astype(np.float32)
224 grad: grad_val,
232 grad: grad_val,
|
| /external/tensorflow/tensorflow/core/kernels/ |
| sparse_conditional_accumulator.h | 162 const Tensor* grad_val = std::get<1>(*grad); variable
176 ctx->allocate_persistent(dtype_, grad_val->shape(), accum_val_persistent_,
180 grad_val->flat<T>();
198 const Tensor* grad_val = std::get<1>(*grad); variable
258 TensorShape sum_shape = grad_val->shape();
266 auto grad_flat = grad_val->flat_outer_dims<T>();
|
| /external/tensorflow/tensorflow/python/training/ |
| training_ops_test.py | 215 grad_val = [np.arange(10), np.arange(10)]
216 grad = np.array(grad_val).astype(dtype)
228 grad_val = [[1.5], [2.5]]
229 grad = np.array(grad_val).astype(dtype)
243 grad_val = [[1.5], [2.5]]
244 grad = np.array(grad_val).astype(dtype)
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| tensor_array_ops_test.py | 870 grad_val = -np.arange(3 * 5, dtype=np_dtype).reshape(3, 5)
872 v0_grad = gradients_impl.gradients([vout], [v0], [grad_val])[0]
873 state0_grad = gradients_impl.gradients([vout], [state0], [grad_val])[0]
874 var_grad = gradients_impl.gradients([vout], [var], [grad_val])[0]
878 v0_grad, state0_grad, var_grad = grad_fn(v0, state0, var, dy=grad_val)
[all...] |
| /external/tensorflow/tensorflow/python/framework/ |
| importer_test.py | [all...] |
| /external/tensorflow/tensorflow/python/ops/ |
| nn_fused_batchnorm_test.py | 284 for grad_val, grad_internal_val in zip(grad_vals, grad_internal_vals):
285 self.assertAllClose(grad_val, grad_internal_val, atol=err_tolerance)
|
| control_flow_ops.py | [all...] |
