Searched refs:input_size_per_batch (Results 1 – 1 of 1) sorted by relevance
969 const int64 input_size_per_batch = grad_in.NumElements() / batch_size; in launch() local982 const int input_start = start * input_size_per_batch; in launch()983 const int input_end = limit * input_size_per_batch; in launch()987 const int64 cur_batch = index / input_size_per_batch; in launch()