Searched refs:output_inv_stddev_ (Results 1 – 2 of 2) sorted by relevance
87 output_inv_stddev_(output_inv_stddev) {} in CudnnBatchNormForwardTrainingThunk()99 buffer_allocations.GetDeviceAddress(output_inv_stddev_)); in ExecuteOnStream()
101 BufferAllocation::Slice output_inv_stddev_; variable