Searched defs:grad_output (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | resampler_ops.cc | 251 XlaOp CalculateGradData(XlaOpKernelContext* ctx, XlaOp grad_output, XlaOp ratio, in CalculateGradData()
362 XlaOp CalculateGradWarp(XlaOpKernelContext* ctx, XlaOp grad_output, XlaOp ratio, in CalculateGradWarp()
643 XlaOp grad_output = ctx->Input("grad_output"); in Compile() local
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | bfloat16_test.cc | 131 auto grad_output = ConstantR4FromArray4D<bfloat16>( in XLA_TEST_F() local
|
| D | batch_normalization_test.cc | 400 auto grad_output = ConstantR4FromArray4D<float>( in XLA_TEST_F() local
431 auto grad_output = ConstantR4FromArray4D<Eigen::half>( in XLA_TEST_F() local
|
| /external/tensorflow/tensorflow/c/ |
| D | while_loop_test.cc | 432 TF_Output grad_output; in TEST_F() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | batchnorm_expander.cc | 420 HloInstruction* grad_output = batch_norm->mutable_operand(4); in HandleBatchNormGrad() local
|
| D | hlo_instruction.cc | 1644 HloInstruction* grad_output, float epsilon, in CreateBatchNormGrad()
|
| D | hlo_instructions.cc | 184 HloInstruction* mean, HloInstruction* variance, HloInstruction* grad_output, in HloBatchNormGradInstruction()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | nn_grad.py | 1228 def _IsotonicRegressionGrad(op, grad_output, grad_segments): argument
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | mlir_hlo_to_hlo.cc | 1418 xla::XlaOp operand, scale, mean, variance, grad_output; in ExportXlaOp() local
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 2843 XlaOp batch_var, XlaOp grad_output, in BatchNormGrad()
4990 const XlaOp grad_output, float epsilon, in BatchNormGrad()
|