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