| /external/tensorflow/tensorflow/compiler/tests/ |
| fused_batchnorm_test.py | 57 # grad_offset = sum(output_y)
70 grad_offset = np.sum(grad_y, axis=(0, 1, 2))
71 return grad_x, grad_scale, grad_offset
229 grad_x, grad_scale, grad_offset, _, _ = gen_nn_ops.fused_batch_norm_grad(
239 [grad_x, grad_scale, grad_offset], {
287 grad_x, grad_scale, grad_offset, _, _ = out
293 [grad_x, grad_scale, grad_offset], {
|
| /external/tensorflow/tensorflow/python/ops/ |
| nn_grad.py | 855 grad_offset: gradient for offset, which is sum(grad_y) in training mode;
931 A tuple (grad_x, grad_scale, grad_offset), where grad_x is the gradient
932 for x, grad_scale the gradient for scale, and grad_offset the gradient
966 grad_offset = math_ops.reduce_sum(grad_y, axis=reduce_axis)
967 return math_ops.cast(grad_x, x_dtype), grad_scale, grad_offset
[all...] |
| nn_fused_batchnorm_test.py | 275 grad_x, grad_scale, grad_offset = gradients_impl.gradients(
281 grad_vals = self.evaluate([grad_x, grad_scale, grad_offset])
294 grad_y, x_shape, grad_offset, scale_shape)
323 grad_y, grad_y32, x_shape, grad_offset, grad_offset32, scale_shape)
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| gpu_layout_assignment_test.cc | 285 auto* grad_offset = builder.AddInstruction( local
297 {operand, scale, mean, var, grad_offset, epsilon,
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| batch_normalization_test.cc | 752 auto grad_offset = local
762 auto I2 = *ReferenceUtil::Broadcast1DTo4D(grad_offset, bounds, feature_index);
[all...] |
| /external/deqp/external/openglcts/modules/glesext/texture_cube_map_array/ |
| esextcTextureCubeMapArraySampling.cpp | 1043 const glw::GLuint grad_offset = i * n_grad_components; local
1051 const glw::GLuint grad_offset = i * n_grad_components; local
1060 const glw::GLuint grad_offset = i * n_grad_components; local
1082 const glw::GLuint grad_offset = i * n_grad_components; local
1090 const glw::GLuint grad_offset = i * n_grad_components; local
1098 const glw::GLuint grad_offset = i * n_grad_components; local
4075 const glw::GLuint grad_offset = layer * n_grad_step_per_layer + face * n_grad_step_per_face; local
[all...] |
