Searched refs:batch_id (Results 1 – 8 of 8) sorted by relevance
52 for (int batch_id = start; batch_id < limit; ++batch_id) { in operator ()() local59 output[batch_id * output_batch_stride + sample_id * data_channels + in operator ()()67 ? data[batch_id * data_batch_stride + in operator ()()73 const T x = warp[batch_id * warp_batch_stride + sample_id * 2]; in operator ()()74 const T y = warp[batch_id * warp_batch_stride + sample_id * 2 + 1]; in operator ()()229 for (int batch_id = start; batch_id < limit; ++batch_id) { in operator ()() local238 ? data[batch_id * data_batch_stride + in operator ()()248 grad_data[batch_id * data_batch_stride + in operator ()()255 grad_warp[batch_id * warp_batch_stride + sample_id * 2 + channel] += in operator ()()260 const T x = warp[batch_id * warp_batch_stride + sample_id * 2]; in operator ()()[all …]
35 data[batch_id * data_batch_stride + data_channels * (y * data_width + x) + \58 const int batch_id = index / output_batch_stride; in Resampler2DKernel() local64 const T x = warp[batch_id * warp_batch_stride + sample_id * 2]; in Resampler2DKernel()65 const T y = warp[batch_id * warp_batch_stride + sample_id * 2 + 1]; in Resampler2DKernel()139 atomicAdd(grad_data + (batch_id * data_batch_stride + \164 const int batch_id = index / output_batch_stride; in ResamplerGrad2DKernel() local170 const int warp_id_x = batch_id * warp_batch_stride + sample_id * 2; in ResamplerGrad2DKernel()
92 const int64 batch_id = offset / samples_per_batch; in TruncatedNormalKernel() local93 T mean = means[single_mean ? 0 : batch_id]; in TruncatedNormalKernel()94 const T input_stddev = stddevs[single_stddev ? 0 : batch_id]; in TruncatedNormalKernel()95 T minval = minvals[single_minval ? 0 : batch_id]; in TruncatedNormalKernel()96 T maxval = maxvals[single_maxval ? 0 : batch_id]; in TruncatedNormalKernel()
91 const int batch_id = blockIdx.x * blockDim.x + threadIdx.x; in lstm_gates() local97 if (batch_id >= batch_size || act_id >= cell_size) return; in lstm_gates()142 const int gid = batch_id * cell_size * 4 + act_id; in lstm_gates()143 const int cid = batch_id * cell_size + act_id; in lstm_gates()304 const int batch_id = blockIdx.x * blockDim.x + threadIdx.x; in lstm_gates_bprop() local307 if (batch_id >= batch_size || act_id >= cell_size) return; in lstm_gates_bprop()309 const int gid = batch_id * cell_size * 4 + act_id; in lstm_gates_bprop()310 const int cid = batch_id * cell_size + act_id; in lstm_gates_bprop()
263 self.batch_id = batch_id = array_ops.placeholder(dtypes.int32)276 batch = state_ops.assign(embedding[batch_id:batch_id + step],441 feed_dict.update({self.batch_id: i, self.step: step})
782 unsigned int batch_id; in tncs_process_if_tnccs() local811 batch_id = atoi(pos); in tncs_process_if_tnccs()813 batch_id); in tncs_process_if_tnccs()814 if (batch_id != tncs->last_batchid + 1) { in tncs_process_if_tnccs()817 batch_id, tncs->last_batchid + 1); in tncs_process_if_tnccs()821 tncs->last_batchid = batch_id; in tncs_process_if_tnccs()
690 unsigned int batch_id; in tncc_process_if_tnccs() local723 batch_id = atoi(pos); in tncc_process_if_tnccs()725 batch_id); in tncc_process_if_tnccs()726 if (batch_id != tncc->last_batchid + 1) { in tncc_process_if_tnccs()729 batch_id, tncc->last_batchid + 1); in tncc_process_if_tnccs()733 tncc->last_batchid = batch_id; in tncc_process_if_tnccs()
838 for batch_id in range(num_batches):