Searched refs:output_tuple_ (Results 1 – 2 of 2) sorted by relevance
152 output_tuple_(output_tuple) { in CudnnBatchNormForwardTrainingThunk()205 buffer_allocations.GetDeviceAddress(output_tuple_)); in ExecuteOnStream()234 output_tuple_(output_tuple) { in CudnnBatchNormBackwardThunk()281 buffer_allocations.GetDeviceAddress(output_tuple_)); in ExecuteOnStream()
107 BufferAllocation::Slice output_tuple_; variable143 BufferAllocation::Slice output_tuple_; variable