Searched refs:grad_in_flat (Results 1 – 1 of 1) sorted by relevance
974 auto grad_in_flat = grad_in.flat<T>(); in launch() local993 grad_out_flat(grad_out_index) += grad_in_flat(index); in launch()