Home
last modified time | relevance | path

Searched refs:in_channel (Results 1 – 20 of 20) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dconv.h78 for (int in_channel = 0; in_channel < input_depth; ++in_channel) { in Conv() local
80 in_x, in_channel)]; in Conv()
82 filter_shape, out_channel, filter_y, filter_x, in_channel)]; in Conv()
161 for (int in_channel = 0; in_channel < input_depth; ++in_channel) { in Conv() local
163 in_x, in_channel)]; in Conv()
165 filter_shape, out_channel, filter_y, filter_x, in_channel)]; in Conv()
229 for (int in_channel = 0; in_channel < input_depth; ++in_channel) { in HybridConvPerChannel() local
238 input_shape, batch, in_y, in_x, in_channel)]; in HybridConvPerChannel()
241 filter_x, in_channel)]; in HybridConvPerChannel()
Dtranspose_conv.h70 for (int in_channel = 0; in_channel < input_depth; ++in_channel) { in TransposeConv() local
85 input_shape, batch, in_y, in_x, in_channel)]; in TransposeConv()
88 filter_x, in_channel)]; in TransposeConv()
161 for (int in_channel = 0; in_channel < input_depth; ++in_channel) { in TransposeConv() local
176 input_shape, batch, in_y, in_x, in_channel)]; in TransposeConv()
179 filter_x, in_channel)]; in TransposeConv()
Dconv3d.h83 for (int in_channel = 0; in_channel < input_num_channels; in Conv3D() local
84 ++in_channel) { in Conv3D()
86 input_shape, batch, in_d, in_y, in_x, in_channel)]; in Conv3D()
89 filter_x, in_channel, out_channel)]; in Conv3D()
Ddepthwiseconv_uint8.h219 for (int in_channel = 0; in_channel < input_depth; ++in_channel) { in RunPerChannel() local
221 const int output_channel = m + in_channel * depth_multiplier; in RunPerChannel()
237 input_shape, batch, in_y, in_x, in_channel)]; in RunPerChannel()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
Ddepthwise_conv.h64 for (int in_channel = 0; in_channel < input_depth; ++in_channel) { in DepthwiseConvPerChannel() local
66 const int output_channel = m + in_channel * depth_multiplier; in DepthwiseConvPerChannel()
81 input_shape, batch, in_y, in_x, in_channel)]; in DepthwiseConvPerChannel()
161 for (int in_channel = 0; in_channel < input_depth; ++in_channel) { in DepthwiseConvPerChannel() local
163 const int output_channel = m + in_channel * depth_multiplier; in DepthwiseConvPerChannel()
178 input_shape, batch, in_y, in_x, in_channel)]; in DepthwiseConvPerChannel()
245 for (int in_channel = 0; in_channel < input_depth; ++in_channel) { in DepthwiseConvHybridPerChannel() local
247 const int output_channel = m + in_channel * depth_multiplier; in DepthwiseConvHybridPerChannel()
262 input_shape, batch, in_y, in_x, in_channel)]; in DepthwiseConvHybridPerChannel()
Dconv.h85 for (int in_channel = 0; in_channel < input_depth; ++in_channel) { in ConvPerChannel() local
87 in_x, in_channel)]; in ConvPerChannel()
89 filter_shape, out_channel, filter_y, filter_x, in_channel)]; in ConvPerChannel()
188 for (int in_channel = 0; in_channel < input_depth; ++in_channel) { in ConvPerChannel() local
190 in_x, in_channel)]; in ConvPerChannel()
192 filter_shape, out_channel, filter_y, filter_x, in_channel)]; in ConvPerChannel()
Dtranspose_conv.h69 for (int in_channel = 0; in_channel < input_depth; ++in_channel) { in TransposeConv() local
84 input_shape, batch, in_y, in_x, in_channel)]; in TransposeConv()
87 filter_x, in_channel)]; in TransposeConv()
166 for (int in_channel = 0; in_channel < input_depth; ++in_channel) { in TransposeConv() local
181 input_shape, batch, in_y, in_x, in_channel)]; in TransposeConv()
184 filter_x, in_channel)]; in TransposeConv()
/external/crosvm/devices/src/
Dserial.rs80 in_channel: Option<Receiver<u8>>, field
104 in_channel: None, in new()
182 self.in_channel = Some(recv_channel); in spawn_input_thread()
191 let in_channel = match self.in_channel.as_ref() { in handle_input_thread() localVariable
195 match in_channel.try_recv() { in handle_input_thread()
201 self.in_channel = None; in handle_input_thread()
/external/tensorflow/tensorflow/lite/micro/kernels/vexriscv/
Ddepthwise_conv.cc105 const int in_channel = begin_ch + offset_ch; in DepthwiseConvPerChannel() local
106 const int output_channel = m + in_channel * depth_multiplier; in DepthwiseConvPerChannel()
109 input_shape, batch, in_y, in_x, in_channel)]; in DepthwiseConvPerChannel()
135 const int in_channel = begin_ch + offset_ch; in DepthwiseConvPerChannel() local
136 const int output_channel = m + in_channel * depth_multiplier; in DepthwiseConvPerChannel()
228 const int in_channel = begin_ch + offset_ch; in DepthwiseConv() local
229 const int output_channel = m + in_channel * depth_multiplier; in DepthwiseConv()
232 input_shape, batch, in_y, in_x, in_channel)]; in DepthwiseConv()
243 const int in_channel = begin_ch + offset_ch; in DepthwiseConv() local
244 const int output_channel = m + in_channel * depth_multiplier; in DepthwiseConv()
/external/tensorflow/tensorflow/python/ops/
Dconv2d_benchmark.py189 dtype, data_format, in_channel, out_channel, hw_stride, padding = args
196 ishape = [batch_size, h, w, in_channel]
199 ishape = [batch_size, in_channel, h, w]
203 fshape = [fh, fw, in_channel, out_channel]
/external/tensorflow/tensorflow/core/kernels/
Ddepthwise_conv_op_gpu.h114 const int in_channel = out_channel / depth_multiplier;
138 in_channel +
143 (in_channel + in_depth * (filter_col + filter_offset_temp));
161 in_channel +
166 (in_channel + in_depth * (filter_col + filter_offset_temp));
370 const int in_channel = out_channel / depth_multiplier;
398 (batch * in_depth + in_channel) * (in_height * in_width);
428 (in_channel + in_depth * (filter_col + filter_offset_temp));
456 (in_channel + in_depth * (filter_col + filter_offset_temp));
829 const int in_channel = thread_id % in_depth;
[all …]
Dconv_ops_using_gemm.cc159 for (int in_channel = 0; in_channel < input_depth; in operator ()() local
160 ++in_channel) { in operator ()()
172 (in_x * input_depth) + in_channel]; in operator ()()
180 (in_channel * filter_count) + out_channel]; in operator ()()
Dquantized_conv_ops.cc134 for (int in_channel = 0; in_channel < input_depth; in operator ()() local
135 ++in_channel) { in operator ()()
147 (in_x * input_depth) + in_channel]; in operator ()()
159 (in_channel * filter_count) + out_channel]; in operator ()()
Dconv_ops_fused_image_transform.cc527 for (int in_channel = 0; in operator ()() local
528 in_channel < task_params.input_depth; in operator ()()
529 ++in_channel) { in operator ()()
530 pixel_params.cache_line_pixel[in_channel] = in operator ()()
531 rect.BilinearSample(in_channel, in operator ()()
/external/tensorflow/tensorflow/lite/kernels/internal/
Dconv_per_channel_quantized_16x8_test.cc75 for (int in_channel = 0; in_channel < input_depth; ++in_channel) { in PickOutputMultiplier() local
85 in_x, in_channel)]; in PickOutputMultiplier()
88 filter_x, in_channel)]; in PickOutputMultiplier()
Ddepthwiseconv_per_channel_quantized_16x8_test.cc68 for (int in_channel = 0; in_channel < input_depth; ++in_channel) { in PickOutputMultiplier() local
70 const int output_channel = m + in_channel * depth_multiplier; in PickOutputMultiplier()
85 in_x, in_channel)]; in PickOutputMultiplier()
Ddepthwiseconv_per_channel_quantized_test.cc70 for (int in_channel = 0; in_channel < input_depth; ++in_channel) { in PickOutputMultiplier() local
72 const int output_channel = m + in_channel * depth_multiplier; in PickOutputMultiplier()
87 in_x, in_channel)]; in PickOutputMultiplier()
/external/crosvm/devices/src/virtio/
Dconsole.rs159 let in_channel = match in_channel_opt.as_ref() { in handle_input() localVariable
176 match in_channel.try_recv() { in handle_input()
237 let mut in_channel = self.spawn_input_thread(&in_avail_evt); in run() localVariable
289 self.handle_input(&mut in_channel, &mut receive_queue); in run()
296 self.handle_input(&mut in_channel, &mut receive_queue); in run()
/external/tensorflow/tensorflow/lite/micro/kernels/xtensa/
Ddepthwise_conv.cc110 for (int in_channel = 0; in_channel < input_depth; ++in_channel) { in DepthwiseConvPerChannel() local
112 const int output_channel = m + in_channel * depth_multiplier; in DepthwiseConvPerChannel()
131 (in_channel); in DepthwiseConvPerChannel()
/external/tensorflow/tensorflow/lite/micro/kernels/vexriscv/doc/
DDepthwiseConv2D_int8.md177 for in_ch < in_channel; step_size = 1
183 for in_ch < in_channel; step_size = 1