Home
last modified time | relevance | path

Searched refs:grad_offset (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/tests/
Dfused_batchnorm_test.py70 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/
Dnn_grad.py966 grad_offset = math_ops.reduce_sum(grad_y, axis=reduce_axis)
967 return math_ops.cast(grad_x, x_dtype), grad_scale, grad_offset
978 grad_offset = math_ops.reduce_sum(grad_y, axis=reduce_axis)
983 return math_ops.cast(grad_x, x_dtype), grad_scale, grad_offset
1016 grad_x, grad_scale, grad_offset = _BatchNormGrad(
1020 [grad_x, grad_scale, grad_offset], [grad_y, x, scale], grad_initial)
Dnn_fused_batchnorm_test.py275 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/
Dgpu_layout_assignment_test.cc285 auto* grad_offset = builder.AddInstruction( in TEST_F() local
297 {operand, scale, mean, var, grad_offset, epsilon, in TEST_F()
/external/deqp/external/openglcts/modules/glesext/texture_cube_map_array/
DesextcTextureCubeMapArraySampling.cpp1043 const glw::GLuint grad_offset = i * n_grad_components; in prepareGradXForFace() local
1045 …tNeighbour(texture_coordinates + texture_coordinates_offset, face, 1, width, grad_x + grad_offset); in prepareGradXForFace()
1051 const glw::GLuint grad_offset = i * n_grad_components; in prepareGradXForFace() local
1054 grad_x + grad_offset); in prepareGradXForFace()
1060 const glw::GLuint grad_offset = i * n_grad_components; in prepareGradXForFace() local
1062 vectorSubtractInPlace<3>(grad_x + grad_offset, texture_coordinates + texture_coordinates_offset); in prepareGradXForFace()
1082 const glw::GLuint grad_offset = i * n_grad_components; in prepareGradYForFace() local
1084 …pNeighbour(texture_coordinates + texture_coordinates_offset, face, 1, width, grad_y + grad_offset); in prepareGradYForFace()
1090 const glw::GLuint grad_offset = i * n_grad_components; in prepareGradYForFace() local
1092 …ur(texture_coordinates + texture_coordinates_offset, face, 4 * width, width, grad_y + grad_offset); in prepareGradYForFace()
[all …]
/external/tensorflow/tensorflow/compiler/xla/tests/
Dbatch_normalization_test.cc752 auto grad_offset = in XLA_TEST_P() local
762 auto I2 = *ReferenceUtil::Broadcast1DTo4D(grad_offset, bounds, feature_index); in XLA_TEST_P()
832 LiteralUtil::CreateR1<float>(grad_offset)}); in XLA_TEST_P()
/external/tensorflow/tensorflow/compiler/xla/g3doc/
Doperation_semantics.md166 | `grad_offset` | `XlaOp` | gradient with respect to input |