Searched refs:in_channel (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/ |
| D | conv.h | 73 for (int in_channel = 0; in_channel < input_depth; ++in_channel) { in ConvPerChannel() local
83 in_x, in_channel)]; in ConvPerChannel()
86 filter_x, in_channel)]; in ConvPerChannel()
|
| D | depthwise_conv.h | 65 for (int in_channel = 0; in_channel < input_depth; ++in_channel) { in DepthwiseConvPerChannel() local
67 const int output_channel = m + in_channel * depth_multiplier; in DepthwiseConvPerChannel()
82 in_x, in_channel)]; in DepthwiseConvPerChannel()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | depthwise_conv_op_gpu.h | 101 const int in_channel = out_channel / depth_multiplier; in DepthwiseConv2dGPUKernelNHWC() local
125 in_channel + in DepthwiseConv2dGPUKernelNHWC()
130 (in_channel + in_depth * (filter_col + filter_offset_temp)); in DepthwiseConv2dGPUKernelNHWC()
147 in_channel + in DepthwiseConv2dGPUKernelNHWC()
152 (in_channel + in_depth * (filter_col + filter_offset_temp)); in DepthwiseConv2dGPUKernelNHWC()
350 const int in_channel = out_channel / depth_multiplier; in DepthwiseConv2dGPUKernelNCHW() local
378 (batch * in_depth + in_channel) * (in_height * in_width); in DepthwiseConv2dGPUKernelNCHW()
408 (in_channel + in_depth * (filter_col + filter_offset_temp)); in DepthwiseConv2dGPUKernelNCHW()
435 (in_channel + in_depth * (filter_col + filter_offset_temp)); in DepthwiseConv2dGPUKernelNCHW()
838 const int in_channel = thread_id % in_depth;
[all …]
|
| D | conv_ops_using_gemm.cc | 157 for (int in_channel = 0; in_channel < input_depth; in operator ()() local
158 ++in_channel) { in operator ()()
170 (in_x * input_depth) + in_channel]; in operator ()()
178 (in_channel * filter_count) + out_channel]; in operator ()()
|
| D | quantized_conv_ops.cc | 134 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 ()()
|
| D | conv_ops_fused_image_transform.cc | 526 for (int in_channel = 0; in operator ()() local
527 in_channel < task_params.input_depth; in operator ()()
528 ++in_channel) { in operator ()()
529 pixel_params.cache_line_pixel[in_channel] = in operator ()()
530 rect.BilinearSample(in_channel, in operator ()()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | conv2d_benchmark.py | 189 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/lite/kernels/internal/reference/ |
| D | reference_ops.h | 174 for (int in_channel = 0; in_channel < input_depth; ++in_channel) { in Conv() local
183 input_shape, batch, in_y, in_x, in_channel)]; in Conv()
186 filter_x, in_channel)]; in Conv()
254 for (int in_channel = 0; in_channel < input_depth; ++in_channel) { in Conv() local
263 in_x, in_channel)]; in Conv()
266 filter_x, in_channel)]; in Conv()
4020 for (int in_channel = 0; in_channel < input_depth; ++in_channel) { in TransposeConv() local
4035 input_shape, batch, in_y, in_x, in_channel)]; in TransposeConv()
4038 filter_x, in_channel)]; in TransposeConv()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 6341 const int in_channel = input_shape.Dims(3); in TransposeConv() local
6343 (filter_shape.FlatSize() * output_shape.FlatSize() * in_channel)) { in TransposeConv()
|