Searched refs:grad_x (Results 1 – 9 of 9) sorted by relevance
79 grad_x = scale * (grad_y - np.mean(grad_y, axis=(0, 1, 2)) -86 return grad_x, grad_scale, grad_offset266 grad_x, grad_scale, grad_offset, _, _ = gen_nn_ops.fused_batch_norm_grad(276 [grad_x, grad_scale, grad_offset], {324 grad_x, grad_scale, grad_offset, _, _ = out330 [grad_x, grad_scale, grad_offset], {
321 grad_x, grad_scale, grad_offset = gradients_impl.gradients(327 grad_vals = self.evaluate([grad_x, grad_scale, grad_offset])336 grad_y, x_shape, grad_x, x_shape)344 x, x_shape, grad_x, x_shape)349 scale, scale_shape, grad_x, x_shape)366 grad_y, grad_y32, x_shape, grad_x, grad_x32, x_shape)374 x, x32, x_shape, grad_x, grad_x32, x_shape)379 scale, scale, scale_shape, grad_x, grad_x32, x_shape)
1001 grad_x = scale * math_ops.rsqrt(var_x + epsilon) * (1008 return math_ops.cast(grad_x, x_dtype), grad_scale, grad_offset1031 grad_x = grad_y * scale * var_rsqrt1032 return math_ops.cast(grad_x, x_dtype), grad_scale, grad_offset1065 grad_x, grad_scale, grad_offset = _BatchNormGrad(1068 grad_grad_y, grad_x, grad_scale = tape.gradient(1069 [grad_x, grad_scale, grad_offset], [grad_y, x, scale], grad_initial)1070 return grad_grad_y, grad_x, grad_scale, None, None1080 grad_grad_y, grad_x, grad_scale, _, _ = _FusedBatchNormGradGrad(op, *grad)1081 return grad_grad_y, grad_x, grad_scale, None, None, None
285 grad_x = sp_sum.values * sp_output.values286 return [None, grad_x, None]
1831 grad_x = math_ops.matmul(grad, y, adjoint_a=False, adjoint_b=True)1834 grad_x = math_ops.matmul(grad, y, adjoint_a=False, adjoint_b=False)1838 grad_x = math_ops.matmul(y, grad, adjoint_a=False, adjoint_b=True)1841 grad_x = math_ops.matmul(y, grad, adjoint_a=True, adjoint_b=True)1844 return grad_x, grad_y1857 grad_x = math_ops.matmul(grad, y, adjoint_a=False, adjoint_b=True)1860 grad_x = math_ops.matmul(grad, y, adjoint_a=False, adjoint_b=False)1864 grad_x = math_ops.matmul(y, grad, adjoint_a=False, adjoint_b=True)1867 grad_x = math_ops.matmul(y, grad, adjoint_a=True, adjoint_b=True)1882 return grad_x, grad_y[all …]
341 grad_x = _GetGradWrt(grad, y, x_shape, x_subs, y_subs, output_subs)346 return grad_x, grad_y361 return grad_x, grad_y366 grad_x = array_ops.reshape(367 math_ops.reduce_sum(grad_x, bx_start + rx), x_shape)370 return grad_x, grad_y
1034 void prepareGradXForFace(glw::GLfloat* grad_x, glw::GLuint face, glw::GLfloat* texture_coordinates,… in prepareGradXForFace() argument1045 …getRightNeighbour(texture_coordinates + texture_coordinates_offset, face, 1, width, grad_x + grad_… in prepareGradXForFace()1054 grad_x + grad_offset); in prepareGradXForFace()1062 vectorSubtractInPlace<3>(grad_x + grad_offset, texture_coordinates + texture_coordinates_offset); in prepareGradXForFace()3552 buffer = &buffers.grad_x; in setupSharedStorageBuffer()4102 …grad_x.init(gl, (glw::GLsizeiptr)(grad_x_buffer_data.size() * sizeof(glw::GLfloat)), &grad_x_buffe…5136 buffer = &buffers.grad_x;
346 bufferDefinition grad_x; member in glcts::TextureCubeMapArraySamplingTest::bufferCollection
4036 grad_x = gradients_impl.gradients(rx, x)[0]4039 self.assertAllClose([156.0, 400.0], sess.run(grad_x, feed_dict=feed_dict))