Searched refs:grad_output_ (Results 1 – 3 of 3) sorted by relevance
228 grad_output_(grad_output), in CudnnBatchNormBackwardThunk()269 buffer_allocations.GetDeviceAddress(grad_output_)), in ExecuteOnStream()
137 BufferAllocation::Slice grad_output_; variable
284 output_, grad_output_ = self.evaluate([output, grad_output])286 self.assert_all_true(np.isfinite(grad_output_[0]))