Searched defs:output_grad_data (Results 1 – 2 of 2) sorted by relevance
219 int64 feature_index, const BufferAllocation::Slice& output_grad_data, in CudnnBatchNormBackwardThunk()259 se::DeviceMemory<float> output_grad_data( in ExecuteOnStream() local
447 auto output_grad_data = assn.GetUniqueSlice(custom_call, {0}).ValueOrDie(); in HandleCustomCall() local