Searched defs:batch_id (Results 1 – 6 of 6) sorted by relevance
58 const int batch_id = index / output_batch_stride; in Resampler2DKernel() local164 const int batch_id = index / output_batch_stride; in ResamplerGrad2DKernel() local
52 for (int batch_id = start; batch_id < limit; ++batch_id) { in operator ()() local229 for (int batch_id = start; batch_id < limit; ++batch_id) { in operator ()() local
91 const int batch_id = blockIdx.x * blockDim.x + threadIdx.x; in lstm_gates() local304 const int batch_id = blockIdx.x * blockDim.x + threadIdx.x; in lstm_gates_bprop() local
92 const int64 batch_id = offset / samples_per_batch; in TruncatedNormalKernel() local
782 unsigned int batch_id; in tncs_process_if_tnccs() local
690 unsigned int batch_id; in tncc_process_if_tnccs() local