HomeSort by relevance Sort by last modified time
    Searched refs:batch_id (Results 1 - 7 of 7) 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 92 const int64 batch_id = offset / samples_per_batch; local
93 T mean = means[single_mean ? 0 : batch_id];
94 const T input_stddev = stddevs[single_stddev ? 0 : batch_id];
95 T minval = minvals[single_minval ? 0 : batch_id];
96 T maxval = maxvals[single_maxval ? 0 : batch_id];
  /external/tensorflow/tensorflow/contrib/rnn/kernels/
lstm_ops_gpu.cu.cc 91 const int batch_id = blockIdx.x * blockDim.x + threadIdx.x; local
97 if (batch_id >= batch_size || act_id >= cell_size) return;
142 const int gid = batch_id * cell_size * 4 + act_id;
143 const int cid = batch_id * cell_size + act_id;
304 const int batch_id = blockIdx.x * blockDim.x + threadIdx.x; local
307 if (batch_id >= batch_size || act_id >= cell_size) return;
309 const int gid = batch_id * cell_size * 4 + act_id;
310 const int cid = batch_id * cell_size + act_id;
  /external/tensorflow/tensorflow/python/keras/
callbacks_v1.py 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})
  /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 313 milliseconds