Home
last modified time | relevance | path

Searched refs:grad_val (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dsparse_conditional_accumulator.h162 const Tensor* grad_val = std::get<1>(*grad); in AllocateAndAssignToAccumGradFunction() local
176 ctx->allocate_persistent(dtype_, grad_val->shape(), accum_val_persistent_, in AllocateAndAssignToAccumGradFunction()
180 grad_val->flat<T>(); in AllocateAndAssignToAccumGradFunction()
198 const Tensor* grad_val = std::get<1>(*grad); in AddToAccumGradFunction() local
258 TensorShape sum_shape = grad_val->shape(); in AddToAccumGradFunction()
266 auto grad_flat = grad_val->flat_outer_dims<T>(); in AddToAccumGradFunction()
/external/tensorflow/tensorflow/compiler/tests/
Dfused_batchnorm_test.py188 grad_val = np.random.random_sample(x_shape).astype(np.float32)
210 x_val, grad_val, scale_val, mean_val, var_val, epsilon, data_format_src)
214 grad_val, data_format_src, data_format)
258 grad_val = np.random.random_sample(x_shape).astype(np.float32)
267 grad_val, data_format_src, data_format)
/external/tensorflow/tensorflow/python/training/
Dtraining_ops_test.py223 grad_val = [np.arange(10), np.arange(10)]
224 grad = np.array(grad_val).astype(dtype)
237 grad_val = [[1.5], [2.5]]
238 grad = np.array(grad_val).astype(dtype)
253 grad_val = [[1.5], [2.5]]
254 grad = np.array(grad_val).astype(dtype)
/external/tensorflow/tensorflow/python/kernel_tests/
Dtensor_array_ops_test.py962 grad_val = -np.arange(3 * 5, dtype=np_dtype).reshape(3, 5)
965 v0_grad, state0_grad, var_grad = grad_fn(v0, state0, var, dy=grad_val)
967 v0_grad = gradients_impl.gradients([vout], [v0], [grad_val])[0]
968 state0_grad = gradients_impl.gradients([vout], [state0], [grad_val])[0]
969 var_grad = gradients_impl.gradients([vout], [var], [grad_val])[0]
999 grad_val[0, :] + grad_val[1, :] + grad_val[2, :],
1000 grad_val[1, :] + grad_val[2, :], grad_val[2, :]
1005 self.assertAllClose(grad_val.sum(axis=0), var_grad_t)
1006 self.assertAllClose(grad_val.sum(axis=0), state0_grad_t)
1428 read_val, grad_val = session.run([r0, grad])
[all …]
Dcontrol_flow_ops_py_test.py1055 grad_val = self.evaluate(grad)
1056 self.assertIsInstance(grad_val, ops.IndexedSlicesValue)
1057 self.assertAllEqual(gradient_checker_v2._to_numpy(grad_val), [[0., 0.],
2810 grad_val = self.evaluate(grad)
2811 self.assertIsInstance(grad_val, ops.IndexedSlicesValue)
2812 arr = gradient_checker_v2._to_numpy(grad_val)
/external/tensorflow/tensorflow/python/framework/
Dimporter_test.py1210 a_val, b_val, grad_val = sess.run([a, b, grad], feed_dict=feed_dict)
1214 self.assertEqual(grad_val, [1.0, 2.0])
1230 a_val, b_val, grad_val = sess.run([a, b, grad], feed_dict=feed_dict)
1233 self.assertEqual(grad_val, [1.0, 2.0])
/external/tensorflow/tensorflow/python/ops/
Dnn_fused_batchnorm_test.py285 for grad_val, grad_internal_val in zip(grad_vals, grad_internal_vals):
286 self.assertAllClose(grad_val, grad_internal_val, atol=err_tolerance)
Dcontrol_flow_ops.py1279 grad_val = constant_op.constant(0, dtype=dtype, shape=shape)
1280 next_grad_val = _NextIteration(grad_val)
1290 grad_val = array_ops.zeros(grad_shape)
1295 next_grad_val = _NextIteration(grad_val)