| /external/tensorflow/tensorflow/compiler/tests/ |
| D | fused_batchnorm_test.py | 85 grad_offset = np.sum(grad_y, axis=(0, 1, 2))
86 return grad_x, grad_scale, grad_offset
266 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, _, _ = out
330 [grad_x, grad_scale, grad_offset], {
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | nn_grad.py | 1007 grad_offset = math_ops.reduce_sum(grad_y, axis=reduce_axis)
1008 return math_ops.cast(grad_x, x_dtype), grad_scale, grad_offset
1027 grad_offset = math_ops.reduce_sum(grad_y, axis=reduce_axis)
1032 return math_ops.cast(grad_x, x_dtype), grad_scale, grad_offset
1065 grad_x, grad_scale, grad_offset = _BatchNormGrad(
1069 [grad_x, grad_scale, grad_offset], [grad_y, x, scale], grad_initial)
|
| D | nn_fused_batchnorm_test.py | 321 grad_x, grad_scale, grad_offset = gradients_impl.gradients(
327 grad_vals = self.evaluate([grad_x, grad_scale, grad_offset])
340 grad_y, x_shape, grad_offset, scale_shape)
370 grad_y, grad_y32, x_shape, grad_offset, grad_offset32, scale_shape)
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_layout_assignment_test.cc | 287 auto* grad_offset = builder.AddInstruction( in TEST_F() local
299 {operand, scale, mean, var, grad_offset, epsilon, in TEST_F()
|
| D | ir_emitter_unnested.cc | 1399 {bn_grad.grad_offset(), "grad_offset"}}, in EmitBatchNormThunkFromMlir()
1421 GetAllocationSliceForMlir(bn_grad.grad_offset())); in EmitBatchNormThunkFromMlir()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/tests/ |
| D | lhlo_gpu_ops.mlir | 7 %grad_offset: memref<8xf32>) -> () {
8 …_grad"(%arg0, %arg1, %arg2, %arg3, %arg4, %grad_operand, %grad_scale, %grad_offset) {epsilon = 1.0…
|
| D | lhlo_ops.mlir | 803 %grad_offset: memref<8xf32>) -> () {
804 …_grad"(%arg0, %arg1, %arg2, %arg3, %arg4, %grad_operand, %grad_scale, %grad_offset) {epsilon = 1.0…
|
| /external/deqp/external/openglcts/modules/glesext/texture_cube_map_array/ |
| D | esextcTextureCubeMapArraySampling.cpp | 1043 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/ |
| D | batch_normalization_test.cc | 850 auto grad_offset = in XLA_TEST_P() local
860 auto I2 = *ReferenceUtil::Broadcast1DTo4D(grad_offset, bounds, feature_index); in XLA_TEST_P()
930 LiteralUtil::CreateR1<float>(grad_offset)}); in XLA_TEST_P()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/IR/ |
| D | lhlo_gpu_ops.td | 57 Arg<LHLO_FpBuffer, "", [MemWrite]>:$grad_offset,
|
| D | lhlo_ops.td | 346 Arg<LHLO_Buffer, "", [MemWrite]>:$grad_offset,
|
| /external/tensorflow/tensorflow/compiler/xla/g3doc/ |
| D | operation_semantics.md | 252 | `grad_offset` | `XlaOp` | gradient with respect to input |
|