HomeSort by relevance Sort by last modified time
    Searched refs:in_channel (Results 1 - 9 of 9) sorted by null

  /external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
conv.h 73 for (int in_channel = 0; in_channel < input_depth; ++in_channel) {
83 in_x, in_channel)];
86 filter_x, in_channel)];
98 // long as the filter size (filter_y * filter_x * in_channel)
depthwise_conv.h 65 for (int in_channel = 0; in_channel < input_depth; ++in_channel) {
67 const int output_channel = m + in_channel * depth_multiplier;
82 in_x, in_channel)];
96 // long as the filter size (filter_y * filter_x * in_channel)
  /external/tensorflow/tensorflow/python/ops/
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/core/kernels/
conv_ops_using_gemm.cc 157 for (int in_channel = 0; in_channel < input_depth;
158 ++in_channel) {
170 (in_x * input_depth) + in_channel];
178 (in_channel * filter_count) + out_channel];
depthwise_conv_op_gpu.h 101 const int in_channel = out_channel / depth_multiplier;
125 in_channel +
130 (in_channel + in_depth * (filter_col + filter_offset_temp));
147 in_channel +
152 (in_channel + in_depth * (filter_col + filter_offset_temp));
350 const int in_channel = out_channel / depth_multiplier;
378 (batch * in_depth + in_channel) * (in_height * in_width);
408 (in_channel + in_depth * (filter_col + filter_offset_temp));
435 (in_channel + in_depth * (filter_col + filter_offset_temp));
    [all...]
quantized_conv_ops.cc 134 for (int in_channel = 0; in_channel < input_depth;
135 ++in_channel) {
147 (in_x * input_depth) + in_channel];
159 (in_channel * filter_count) + out_channel];
conv_ops_fused_image_transform.cc 526 for (int in_channel = 0;
527 in_channel < task_params.input_depth;
528 ++in_channel) {
529 pixel_params.cache_line_pixel[in_channel] =
530 rect.BilinearSample(in_channel,
    [all...]
  /external/tensorflow/tensorflow/lite/kernels/internal/reference/
reference_ops.h 174 for (int in_channel = 0; in_channel < input_depth; ++in_channel) {
183 input_shape, batch, in_y, in_x, in_channel)];
186 filter_x, in_channel)];
254 for (int in_channel = 0; in_channel < input_depth; ++in_channel) {
263 in_x, in_channel)];
266 filter_x, in_channel)];
    [all...]
  /external/tensorflow/tensorflow/lite/kernels/internal/optimized/
optimized_ops.h     [all...]

Completed in 253 milliseconds