Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/contrib/resampler/python/ops/
Dresampler_ops.py65 def _resampler_grad(op, grad_output): argument
/external/tensorflow/tensorflow/compiler/xla/tests/
Dbfloat16_test.cc133 auto grad_output = ConstantR4FromArray4D<bfloat16>( in XLA_TEST_F() local
Dbatch_normalization_test.cc355 auto grad_output = ConstantR4FromArray4D<float>( in XLA_TEST_P() local
/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/service/gpu/
Dcudnn_batchnorm_thunk.cc218 const BufferAllocation::Slice& grad_output, float epsilon, in CudnnBatchNormBackwardThunk()
/external/tensorflow/tensorflow/c/
Dwhile_loop_test.cc433 TF_Output grad_output; in TEST_F() local
/external/tensorflow/tensorflow/contrib/resampler/kernels/
Dresampler_ops_gpu.cu.cc146 const T* __restrict__ grad_output, T* __restrict__ grad_data, in ResamplerGrad2DKernel()
Dresampler_ops.cc338 const ::tensorflow::Tensor& grad_output = ctx->input(2); in Compute() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dbatchnorm_expander.cc468 HloInstruction* grad_output = batch_norm->mutable_operand(4); in HandleBatchNormGrad() local
Dhlo_instruction.cc1084 HloInstruction* grad_output, float epsilon, in CreateBatchNormGrad()
Dhlo_instructions.cc146 HloInstruction* mean, HloInstruction* variance, HloInstruction* grad_output, in HloBatchNormGradInstruction()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc2096 const XlaOp& grad_output, float epsilon, in BatchNormGrad()
3503 const XlaOp& grad_output, float epsilon, in BatchNormGrad()