Searched refs:batch_i (Results 1 – 5 of 5) sorted by relevance
41 Index batch_i = 0; // The batch index into params to use for i. in GatherOpKernel() local57 batch_i = entries_count; in GatherOpKernel()59 batch_i = entries_count / outer_size; in GatherOpKernel()60 outer_i = entries_count - batch_i * outer_size; in GatherOpKernel()67 Index gather_i = ldg(indices + batch_i * indices_size + indices_i); in GatherOpKernel()78 (batch_i * outer_size + outer_i) * gather_dim_size + gather_i in GatherOpKernel()
39 Index batch_i = 0; in GatherOpKernel() local48 batch_i = batch_indices_i / indices_size; in GatherOpKernel()50 indices_i = batch_indices_i - batch_i * indices_size; in GatherOpKernel()68 (batch_i * gather_dim_size + gather_i) * slice_size + slice_i; in GatherOpKernel()
597 for (batch_i, seq_index), vocabulary_id in zip(feature_indices,599 if batch_i == i:
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()
54 for batch_i, batch in enumerate(x):56 x_ix.append([batch_i, time])