| /external/tensorflow/tensorflow/contrib/resampler/kernels/ |
| D | resampler_ops_gpu.cu.cc | 35 data[batch_id * data_batch_stride + data_channels * (y * data_width + x) + \
43 const int data_channels, in Resampler2DKernel() argument
45 const int output_data_size = batch_size * num_sampling_points * data_channels; in Resampler2DKernel()
54 const int data_batch_stride = data_height * data_width * data_channels; in Resampler2DKernel()
56 const int output_batch_stride = num_sampling_points * data_channels; in Resampler2DKernel()
60 const int chan = index_in_batch % data_channels; in Resampler2DKernel()
61 const int sample_id = index_in_batch / data_channels; in Resampler2DKernel()
117 const int data_channels, const int num_sampling_points) { in operator ()()
119 batch_size * num_sampling_points * data_channels; in operator ()()
125 data_channels, num_sampling_points)); in operator ()()
[all …]
|
| D | resampler_ops.cc | 44 const int data_channels, const int num_sampling_points) { in operator ()()
46 const int data_batch_stride = data_height * data_width * data_channels; in operator ()()
47 const int output_batch_stride = num_sampling_points * data_channels; in operator ()()
59 output[batch_id * output_batch_stride + sample_id * data_channels + in operator ()()
68 data_channels * (y * data_width + x) + chan] in operator ()()
93 for (int chan = 0; chan < data_channels; ++chan) { in operator ()()
103 for (int chan = 0; chan < data_channels; ++chan) { in operator ()()
116 static_cast<int64>(num_sampling_points) * data_channels * 1000; in operator ()()
162 const int data_channels = data_shape.dim_size(3); in Compute() local
164 output_shape.set_dim(output_shape.dims() - 1, data_channels); in Compute()
[all …]
|
| D | resampler_ops.h | 37 const int data_channels, const int num_sampling_points);
48 const int data_channels, const int num_sampling_points);
|
| /external/tensorflow/tensorflow/contrib/resampler/python/ops/ |
| D | resampler_ops_test.py | 131 data_channels = 5
137 data_shape = (batch_size, data_height, data_width, data_channels)
145 [None, warp_height, warp_width, data_channels])
151 for c in xrange(data_channels):
163 data_channels = 3
169 data_shape = (batch_size, data_height, data_width, data_channels)
206 data_channels = 5
215 data_channels)
229 data_shape = (batch_size, data_height, data_width, data_channels)
242 data_shape = (batch_size, data_height, data_width, data_channels)
[all …]
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | resampler_ops.cc | 139 int64 data_channels, int warp_dims) { in Gather2by2Neighbors() argument
159 /*slice_sizes=*/{1, 2, 2, data_channels}); in Gather2by2Neighbors()
251 int64 last_warp_dim, int64 data_channels, in CalculateGradData() argument
271 weights_with_channels_dims.push_back(data_channels); in CalculateGradData()
327 /*limit_indices=*/{batch_size, width + 1, height + 1, data_channels}, in CalculateGradData()
361 TensorShape warp_shape, int64 data_channels, in CalculateGradWarp() argument
393 data_channels, warp_shape.dims()); in CalculateGradWarp()
492 const int64 data_channels = data_shape.dim_size(3); in Compile() local
523 ctx->builder(), data, gather_indices, data_channels, warp_shape.dims()); in Compile()
581 result_dims.push_back(data_channels); in Compile()
[all …]
|
| /external/adhd/cras/src/dsp/ |
| D | drc_kernel.c | 970 static void dk_copy_fragment(struct drc_kernel *dk, float *data_channels[], in dk_copy_fragment() argument
979 &data_channels[j][frame_index], in dk_copy_fragment()
981 memcpy(&data_channels[j][frame_index], in dk_copy_fragment()
996 static void dk_process_delay_only(struct drc_kernel *dk, float *data_channels[], in dk_process_delay_only() argument
1014 &data_channels[j][i], in dk_process_delay_only()
1016 memcpy(&data_channels[j][i], in dk_process_delay_only()
1029 void dk_process(struct drc_kernel *dk, float *data_channels[], unsigned count) in dk_process() argument
1035 dk_process_delay_only(dk, data_channels, count); in dk_process()
1048 dk_copy_fragment(dk, data_channels, i, fragment); in dk_process()
|
| D | drc_kernel.h | 111 void dk_process(struct drc_kernel *dk, float *data_channels[], unsigned count);
|