Searched defs:output_grad_scale (Results 1 – 2 of 2) sorted by relevance
220 const BufferAllocation::Slice& output_grad_scale, in CudnnBatchNormBackwardThunk()261 se::DeviceMemory<float> output_grad_scale( in ExecuteOnStream() local
448 auto output_grad_scale = assn.GetUniqueSlice(custom_call, {1}).ValueOrDie(); in HandleCustomCall() local