Searched refs:output_size_per_batch (Results 1 – 1 of 1) sorted by relevance
968 const int64 output_size_per_batch = grad_out->NumElements() / batch_size; in launch() local976 const int64 output_start = start * output_size_per_batch; in launch()977 const int64 output_end = limit * output_size_per_batch; in launch()988 grad_out_index += cur_batch * output_size_per_batch; in launch()