Searched defs:data_channels (Results 1 - 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/contrib/resampler/kernels/
H A Dresampler_ops.cc44 const int data_channels, const int num_sampling_points) {
46 const int data_batch_stride = data_height * data_width * data_channels;
47 const int output_batch_stride = num_sampling_points * data_channels;
59 output[batch_id * output_batch_stride + sample_id * data_channels +
68 data_channels * (y * data_width + x) + chan]
93 for (int chan = 0; chan < data_channels; ++chan) {
103 for (int chan = 0; chan < data_channels; ++chan) {
116 static_cast<int64>(num_sampling_points) * data_channels * 1000;
140 "data_width, data_channels], but is: ",
162 const int data_channels variable
40 operator ()(::tensorflow::OpKernelContext* ctx, const CPUDevice& d, const T* __restrict__ data, const T* __restrict__ warp, T* __restrict__ output, const int batch_size, const int data_height, const int data_width, const int data_channels, const int num_sampling_points) argument
206 operator ()(::tensorflow::OpKernelContext* ctx, const CPUDevice& d, const T* __restrict__ data, const T* __restrict__ warp, const T* __restrict__ grad_output, T* __restrict__ grad_data, T* __restrict__ grad_warp, const int batch_size, const int data_height, const int data_width, const int data_channels, const int num_sampling_points) argument
351 const int data_channels = data_shape.dim_size(3); variable
[all...]
H A Dresampler_ops_gpu.cu.cc35 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_point
39 Resampler2DKernel(const T* __restrict__ data, const T* __restrict__ warp, T* __restrict__ output, const int batch_size, const int data_height, const int data_width, const int data_channels, const int num_sampling_points) argument
113 operator ()(::tensorflow::OpKernelContext* ctx, const GPUDevice& d, const T* __restrict__ data, const T* __restrict__ warp, T* __restrict__ output, const int batch_size, const int data_height, const int data_width, const int data_channels, const int num_sampling_points) argument
144 ResamplerGrad2DKernel( const T* __restrict__ data, const T* __restrict__ warp, const T* __restrict__ grad_output, T* __restrict__ grad_data, T* __restrict__ grad_warp, const int batch_size, const int data_height, const int data_width, const int data_channels, const int num_sampling_points) argument
243 operator ()(::tensorflow::OpKernelContext* ctx, const GPUDevice& d, const T* __restrict__ data, const T* __restrict__ warp, const T* __restrict__ grad_output, T* __restrict__ grad_data, T* __restrict__ grad_warp, const int batch_size, const int data_height, const int data_width, const int data_channels, const int num_sampling_points) argument
[all...]
/external/adhd/cras/src/dsp/
H A Ddrc_kernel.c970 static void dk_copy_fragment(struct drc_kernel *dk, float *data_channels[], argument
979 &data_channels[j][frame_index],
981 memcpy(&data_channels[j][frame_index],
996 static void dk_process_delay_only(struct drc_kernel *dk, float *data_channels[], argument
1014 &data_channels[j][i],
1016 memcpy(&data_channels[j][i],
1029 void dk_process(struct drc_kernel *dk, float *data_channels[], unsigned count) argument
1035 dk_process_delay_only(dk, data_channels, count);
1048 dk_copy_fragment(dk, data_channels, i, fragment);

Completed in 124 milliseconds