Searched refs:batch_i (Results 1 – 3 of 3) sorted by relevance
37 Index batch_i = 0; in GatherOpKernel() local46 batch_i = batch_indices_i / indices_size; in GatherOpKernel()48 indices_i = batch_indices_i - batch_i * indices_size; in GatherOpKernel()66 (batch_i * gather_dim_size + gather_i) * slice_size + slice_i; in GatherOpKernel()
1099 int batch_i; in plot_io_movie() local1167 batch_i = 0; in plot_io_movie()1168 while (i < cols && batch_i < batch_count) { in plot_io_movie()1182 batch_i++; in plot_io_movie()
50 for batch_i, batch in enumerate(x):52 x_ix.append([batch_i, time])