Lines Matching defs:data_channels
35 data[batch_id * data_batch_stride + data_channels * (y * data_width + x) + \
43 const int data_channels,
45 const int output_data_size = batch_size * num_sampling_points * data_channels;
54 const int data_batch_stride = data_height * data_width * data_channels;
56 const int output_batch_stride = num_sampling_points * data_channels;
60 const int chan = index_in_batch % data_channels;
61 const int sample_id = index_in_batch / data_channels;
117 const int data_channels, const int num_sampling_points) {
119 batch_size * num_sampling_points * data_channels;
125 data_channels, num_sampling_points);
140 data_channels * (y * data_width + x) + chan), \
148 const int data_width, const int data_channels,
151 batch_size * num_sampling_points * data_channels;
160 const int data_batch_stride = data_height * data_width * data_channels;
162 const int output_batch_stride = num_sampling_points * data_channels;
166 const int chan = index_in_batch % data_channels;
167 const int sample_id = index_in_batch / data_channels;
248 const int data_channels, const int num_sampling_points) {
253 batch_size * data_height * data_width * data_channels;
267 batch_size * num_sampling_points * data_channels;
272 data_height, data_width, data_channels, num_sampling_points);