Home
last modified time | relevance | path

Searched refs:grad_y (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/python/ops/
Dgradients_util.py218 grad_y = grad_ys[i]
221 if grad_y is None:
232 if not grad_y.dtype.is_floating and not grad_y.dtype.is_integer:
236 "real or integer" % (dtypes.as_dtype(grad_y.dtype).name, y,
239 if not grad_y.dtype.is_complex:
243 grad_y.dtype).name, y, dtypes.as_dtype(y.dtype).name))
245 if grad_y.dtype != dtypes.variant:
249 grad_y.dtype).name, y, dtypes.as_dtype(y.dtype).name))
253 if grad_y.dtype == dtypes.resource:
255 "be a resource" % (grad_y, y))
[all …]
Dnn_grad.py859 grad_y = grad[0]
869 grad_y,
882 grad_y = array_ops.transpose(grad_y, [0, 2, 3, 1])
884 grad_y,
907 def _BatchNormGrad(grad_y, argument
940 grad_y = math_ops.cast(grad_y, dtypes.float32)
950 mean_grad_y = math_ops.reduce_mean(grad_y, reduce_axis, keepdims=keepdims)
956 grad_y_offset = grad_y - mean_grad_y
959 grad_y * x_offset, axis=reduce_axis, keepdims=keepdims)
963 grad_y * x_offset, axis=reduce_axis, keepdims=keepdims)
[all …]
Dnn_fused_batchnorm_test.py258 grad_y = constant_op.constant(grad_y_val, name='grad_y')
276 y, [x, scale, offset], grad_y)
282 grad_internal = nn_grad._BatchNormGrad(grad_y, x, scale, pop_mean,
290 grad_y, x_shape, grad_x, x_shape)
292 grad_y, x_shape, grad_scale, scale_shape)
294 grad_y, x_shape, grad_offset, scale_shape)
319 grad_y, grad_y32, x_shape, grad_x, grad_x32, x_shape)
321 grad_y, grad_y32, x_shape, grad_scale, grad_scale32, scale_shape)
323 grad_y, grad_y32, x_shape, grad_offset, grad_offset32, scale_shape)
Dcond_v2.py281 for y, grad_y in zip(func_graph.outputs, grads):
285 grad_ys.append(grad_y)
Dmath_grad.py1255 grad_y = math_ops.matmul(x, grad, adjoint_a=True, adjoint_b=False)
1258 grad_y = math_ops.matmul(grad, x, adjoint_a=True, adjoint_b=False)
1262 grad_y = math_ops.matmul(x, grad, adjoint_a=False, adjoint_b=False)
1265 grad_y = math_ops.matmul(grad, x, adjoint_a=True, adjoint_b=True)
1267 return grad_x, grad_y
/external/tensorflow/tensorflow/compiler/tests/
Dfused_batchnorm_test.py52 def _reference_grad(self, x, grad_y, scale, mean, var, epsilon, data_format): argument
64 grad_x = scale * (grad_y - np.mean(grad_y, axis=(0, 1, 2)) -
65 (x - mean) * np.mean(grad_y *
69 grad_y * (x - mean) / np.sqrt(var + epsilon), axis=(0, 1, 2))
70 grad_offset = np.sum(grad_y, axis=(0, 1, 2))
/external/deqp/external/openglcts/modules/glesext/texture_cube_map_array/
DesextcTextureCubeMapArraySampling.cpp1073 void prepareGradYForFace(glw::GLfloat* grad_y, glw::GLuint face, glw::GLfloat* texture_coordinates,… in prepareGradYForFace() argument
1084 …getTopNeighbour(texture_coordinates + texture_coordinates_offset, face, 1, width, grad_y + grad_of… in prepareGradYForFace()
1092 …getTopNeighbour(texture_coordinates + texture_coordinates_offset, face, 4 * width, width, grad_y +… in prepareGradYForFace()
1100 vectorSubtractInPlace<3>(grad_y + grad_offset, texture_coordinates + texture_coordinates_offset); in prepareGradYForFace()
3555 buffer = &buffers.grad_y; in setupSharedStorageBuffer()
4103grad_y.init(gl, (glw::GLsizeiptr)(grad_y_buffer_data.size() * sizeof(glw::GLfloat)), &grad_y_buffe…
5140 buffer = &buffers.grad_y;
DesextcTextureCubeMapArraySampling.hpp347 bufferDefinition grad_y; member in glcts::TextureCubeMapArraySamplingTest::bufferCollection
/external/tensorflow/tensorflow/python/kernel_tests/
Dcwise_ops_test.py1156 grad_y = constant_op.constant(0, dtype=dtype)
1158 g = gradients_impl.gradients(y, args, grad_ys=grad_y)