HomeSort by relevance Sort by last modified time
    Searched refs:batch_id (Results 1 - 6 of 6) sorted by null

  /external/tensorflow/tensorflow/contrib/resampler/kernels/
resampler_ops.cc 52 for (int batch_id = start; batch_id < limit; ++batch_id) {
59 output[batch_id * output_batch_stride + sample_id * data_channels +
67 ? data[batch_id * data_batch_stride +
73 const T x = warp[batch_id * warp_batch_stride + sample_id * 2];
74 const T y = warp[batch_id * warp_batch_stride + sample_id * 2 + 1];
229 for (int batch_id = start; batch_id < limit; ++batch_id) {
    [all...]
resampler_ops_gpu.cu.cc 35 data[batch_id * data_batch_stride + data_channels * (y * data_width + x) + \
51 // index = batch_id * num_sampling_points * num_chans +
58 const int batch_id = index / output_batch_stride; local
64 const T x = warp[batch_id * warp_batch_stride + sample_id * 2];
65 const T y = warp[batch_id * warp_batch_stride + sample_id * 2 + 1];
139 atomicAdd(grad_data + (batch_id * data_batch_stride + \
157 // index = batch_id * num_sampling_points * num_chans +
164 const int batch_id = index / output_batch_stride; local
170 const int warp_id_x = batch_id * warp_batch_stride + sample_id * 2;
  /external/tensorflow/tensorflow/core/kernels/
parameterized_truncated_normal_op_gpu.cu.cc 79 const int64 batch_id = offset / samples_per_batch; local
80 T mean = means[single_mean ? 0 : batch_id];
81 const T input_stddev = stddevs[single_stddev ? 0 : batch_id];
82 T minval = minvals[single_minval ? 0 : batch_id];
83 T maxval = maxvals[single_maxval ? 0 : batch_id];
  /external/tensorflow/tensorflow/contrib/rnn/kernels/
lstm_ops_gpu.cu.cc 48 const int batch_id = blockIdx.x * blockDim.x + threadIdx.x; local
51 if (batch_id >= batch_size || act_id >= cell_size) return;
96 const int gid = batch_id * cell_size * 4 + act_id;
97 const int cid = batch_id * cell_size + act_id;
254 const int batch_id = blockIdx.x * blockDim.x + threadIdx.x; local
257 if (batch_id >= batch_size || act_id >= cell_size) return;
259 const int gid = batch_id * cell_size * 4 + act_id;
260 const int cid = batch_id * cell_size + act_id;
  /external/wpa_supplicant_8/src/eap_peer/
tncc.c 690 unsigned int batch_id; local
723 batch_id = atoi(pos);
725 batch_id);
726 if (batch_id != tncc->last_batchid + 1) {
729 batch_id, tncc->last_batchid + 1);
733 tncc->last_batchid = batch_id;
  /external/wpa_supplicant_8/src/eap_server/
tncs.c 782 unsigned int batch_id; local
811 batch_id = atoi(pos);
813 batch_id);
814 if (batch_id != tncs->last_batchid + 1) {
817 batch_id, tncs->last_batchid + 1);
821 tncs->last_batchid = batch_id;

Completed in 224 milliseconds