Home
last modified time | relevance | path

Searched defs:grad_output (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_batchnorm_thunk.cc117 const BufferAllocation::Slice& grad_output, in CudnnBatchNormBackwardThunk()
138 se::DeviceMemoryBase grad_output = in ExecuteOnStream() local
Dcudnn_batchnorm_runner.cc58 se::DeviceMemoryBase grad_output; member
287 se::DeviceMemoryBase output_grad_data, se::DeviceMemoryBase grad_output, in RunCudnnBatchNormBackward()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dresampler_ops.cc248 XlaOp CalculateGradData(XlaOpKernelContext* ctx, XlaOp grad_output, XlaOp ratio, in CalculateGradData()
359 XlaOp CalculateGradWarp(XlaOpKernelContext* ctx, XlaOp grad_output, XlaOp ratio, in CalculateGradWarp()
645 XlaOp grad_output = ctx->Input("grad_output"); in Compile() local
/external/tensorflow/tensorflow/compiler/xla/tests/
Dbfloat16_test.cc133 auto grad_output = ConstantR4FromArray4D<bfloat16>( in XLA_TEST_F() local
Dbatch_normalization_test.cc416 auto grad_output = ConstantR4FromArray4D<float>( in XLA_TEST_P() local
447 auto grad_output = ConstantR4FromArray4D<Eigen::half>( in XLA_TEST_P() local
/external/tensorflow/tensorflow/c/
Dwhile_loop_test.cc432 TF_Output grad_output; in TEST_F() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dbatchnorm_expander.cc417 HloInstruction* grad_output = batch_norm->mutable_operand(4); in HandleBatchNormGrad() local
Dhlo_instruction.cc1522 HloInstruction* grad_output, float epsilon, in CreateBatchNormGrad()
Dhlo_instructions.cc152 HloInstruction* mean, HloInstruction* variance, HloInstruction* grad_output, in HloBatchNormGradInstruction()
/external/tensorflow/tensorflow/python/ops/
Dnn_grad.py1187 def _IsotonicRegressionGrad(op, grad_output, grad_segments): argument
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc2717 XlaOp batch_var, XlaOp grad_output, in BatchNormGrad()
4818 const XlaOp grad_output, float epsilon, in BatchNormGrad()