| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cudnn_batchnorm_thunk.cc | 122 const BufferAllocation::Slice& grad_output, in CudnnBatchNormBackwardThunk() argument
132 grad_output_(grad_output), in CudnnBatchNormBackwardThunk()
143 se::DeviceMemoryBase grad_output = in ExecuteOnStream() local
154 config_, operand, output_grad_data, grad_output, output_grad_scale, in ExecuteOnStream()
|
| D | cudnn_batchnorm_runner.cc | 58 se::DeviceMemoryBase grad_output; member
195 se::DeviceMemory<ElemType>(params->grad_output), // in RunCudnnBatchNormBackwardImpl()
286 se::DeviceMemoryBase output_grad_data, se::DeviceMemoryBase grad_output, in RunCudnnBatchNormBackward() argument
294 backward_params.grad_output = grad_output; in RunCudnnBatchNormBackward()
|
| D | cudnn_batchnorm_runner.h | 56 se::DeviceMemoryBase output_grad_data, se::DeviceMemoryBase grad_output,
|
| D | cudnn_batchnorm_thunk.h | 111 const BufferAllocation::Slice& grad_output,
|
| D | ir_emitter_unnested.cc | 1402 {bn_grad.grad_output(), "grad_output"}, in EmitBatchNormThunkFromMlir()
1411 TF_ASSIGN_OR_RETURN(auto grad_output, in EmitBatchNormThunkFromMlir()
1412 GetAllocationSliceForMlir(bn_grad.grad_output())); in EmitBatchNormThunkFromMlir()
1424 config.output_shape = TypeToShape(bn_grad.grad_output().getType()); in EmitBatchNormThunkFromMlir()
1431 /*config=*/get_batch_norm_config(bn_grad, bn_grad.grad_output()), in EmitBatchNormThunkFromMlir()
1436 /*grad_output=*/grad_output, in EmitBatchNormThunkFromMlir()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | resampler_ops.cc | 248 XlaOp CalculateGradData(XlaOpKernelContext* ctx, XlaOp grad_output, XlaOp ratio, in CalculateGradData() argument
296 grad_output, weights_with_channels_dims, grad_output_indices); in CalculateGradData()
359 XlaOp CalculateGradWarp(XlaOpKernelContext* ctx, XlaOp grad_output, XlaOp ratio, in CalculateGradWarp() argument
452 grad_output * weight_y * bottom_right_minus_bottom_left + in CalculateGradWarp()
469 auto y_before_reduce = grad_output * weight_x * bottom_right_minus_top_right + in CalculateGradWarp()
645 XlaOp grad_output = ctx->Input("grad_output"); in Compile() local
659 ctx, grad_output, ratio, gather_indices, warp, warp_type, warp_shape, in Compile()
663 CalculateGradWarp(ctx, grad_output, ratio, gather_indices, data, in Compile()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | bfloat16_test.cc | 133 auto grad_output = ConstantR4FromArray4D<bfloat16>( in XLA_TEST_F() local
140 BatchNormGrad(operand, scale, mean, var, grad_output, in XLA_TEST_F()
|
| D | batch_normalization_test.cc | 416 auto grad_output = ConstantR4FromArray4D<float>( in XLA_TEST_P() local
420 BatchNormGrad(operand, scale, mean, var, grad_output, in XLA_TEST_P()
447 auto grad_output = ConstantR4FromArray4D<Eigen::half>( in XLA_TEST_P() local
451 auto grad_output_f32 = ConvertElementType(grad_output, F32); in XLA_TEST_P()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | batchnorm_expander.cc | 417 HloInstruction* grad_output = batch_norm->mutable_operand(4); in HandleBatchNormGrad() local
470 add_binary(activation_shape, HloOpcode::kMultiply, grad_output, in HandleBatchNormGrad()
484 feature_shape, grad_output, zero, dimensions_without_feature, in HandleBatchNormGrad()
518 auto i1 = add_binary(activation_shape, HloOpcode::kMultiply, grad_output, in HandleBatchNormGrad()
|
| D | hlo_parser_test.cc | 789 …4 (input: f32[2,2,2,2], scale: f32[2], mean: f32[2], variance: f32[2], grad_output: f32[2,2,2,2]) … in CreateTestCases()
794 %grad_output = f32[2,2,2,2]{3,2,1,0} parameter(4) in CreateTestCases()
795 …} %scale, f32[2]{0} %mean, f32[2]{0} %variance, f32[2,2,2,2]{3,2,1,0} %grad_output), epsilon=0.001… in CreateTestCases()
|
| D | hlo_instruction.h | 860 HloInstruction* grad_output, float epsilon, int64 feature_index);
|
| D | hlo_instructions.cc | 149 HloInstruction* mean, HloInstruction* variance, HloInstruction* grad_output, in HloBatchNormGradInstruction() argument
155 AppendOperand(grad_output); in HloBatchNormGradInstruction()
|
| D | hlo_instructions.h | 95 HloInstruction* grad_output, float epsilon, int64 feature_index);
|
| D | hlo_instruction.cc | 1345 HloInstruction* grad_output, float epsilon, in CreateBatchNormGrad() argument
1348 shape, operand, scale, mean, variance, grad_output, epsilon, in CreateBatchNormGrad()
|
| /external/tensorflow/tensorflow/c/ |
| D | while_loop_test.cc | 432 TF_Output grad_output; in TEST_F() local
434 nullptr, s_, &grad_output); in TEST_F()
438 Run({grad_output}, {0}); in TEST_F()
|
| /external/tensorflow/tensorflow/python/kernel_tests/distributions/ |
| D | special_math_test.py | 281 output, grad_output = _value_and_gradient(
284 output_, grad_output_ = self.evaluate([output, grad_output])
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/IR/ |
| D | lhlo_gpu_ops.td | 54 Arg<LHLO_FpBuffer, "", [MemRead]>:$grad_output,
|
| D | lhlo_ops.td | 343 Arg<LHLO_Buffer, "", [MemRead]>:$grad_output,
|
| D | hlo_ops.td | 759 HLO_Tensor:$grad_output,
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | nn_grad.py | 1199 def _IsotonicRegressionGrad(op, grad_output, grad_segments): argument
1213 return _MeanAggregator(grad_output, segments)
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/tests/translate/ |
| D | import.hlotxt | 88 …d (input: f32[2,2,2,2], scale: f32[2], mean: f32[2], variance: f32[2], grad_output: f32[2,2,2,2]) …
93 %grad_output = f32[2,2,2,2] parameter(4)
98 …] %input, f32[2] %scale, f32[2] %mean, f32[2] %variance, f32[2,2,2,2] %grad_output), epsilon=0.001…
|
| D | export.mlir | 75 …scale: tensor<2xf32>, %mean: tensor<2xf32>, %variance: tensor<2xf32>, %grad_output: tensor<2x2x2x2…
76 …%0 = "mhlo.batch_norm_grad" (%input, %scale, %mean, %variance, %grad_output) {epsilon = 0.001 : f3…
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | while_v2_test.py | 1037 grad_output = GetAccumulatorForInputAtIndex(grad_while_op,
1039 _, val = list_ops.tensor_list_pop_back(grad_output,
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.h | 891 XlaOp batch_var, XlaOp grad_output, float epsilon,
1409 XlaOp batch_var, XlaOp grad_output, float epsilon,
2535 XlaOp batch_var, XlaOp grad_output, float epsilon,
|
| D | xla_builder.cc | 2689 XlaOp batch_var, XlaOp grad_output, in BatchNormGrad() argument
2699 GetShapePtr(grad_output)); in BatchNormGrad()
2710 {operand, scale, batch_mean, batch_var, grad_output}); in BatchNormGrad()
4956 const XlaOp grad_output, float epsilon, in BatchNormGrad() argument
4959 grad_output, epsilon, feature_index); in BatchNormGrad()
|