Searched defs:output_inv_stddev (Results 1 – 2 of 2) sorted by relevance
141 const BufferAllocation::Slice& output_inv_stddev, in CudnnBatchNormForwardTrainingThunk()178 se::DeviceMemory<float> output_inv_stddev( in ExecuteOnStream() local
419 auto output_inv_stddev = assn.GetUniqueSlice(custom_call, {2}).ValueOrDie(); in HandleCustomCall() local