Searched refs:grad_out_flat (Results 1 – 1 of 1) sorted by relevance
972 auto grad_out_flat = grad_out->flat<T>(); in launch() local978 EigenMatrixMap inputShard(grad_out_flat.data() + output_start, 1, in launch()993 grad_out_flat(grad_out_index) += grad_in_flat(index); in launch()