Home
last modified time | relevance | path

Searched defs:out_channel (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
Dconv.h67 for (int out_channel = 0; out_channel < output_depth; ++out_channel) { in ConvPerChannel() local
/external/tensorflow/tensorflow/core/kernels/
Ddepthwise_conv_op_gpu.h96 const int out_channel = thread_id % out_depth; in DepthwiseConv2dGPUKernelNHWC() local
344 const int out_channel = (thread_id / out_width / out_height) % out_depth; in DepthwiseConv2dGPUKernelNCHW() local
Dconv_ops_using_gemm.cc132 for (int out_channel = 0; out_channel < filter_count; ++out_channel) { in operator ()() local
Dquantized_conv_ops.cc109 for (int out_channel = 0; out_channel < filter_count; ++out_channel) { in operator ()() local
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dreference_ops.h168 for (int out_channel = 0; out_channel < output_depth; ++out_channel) { in Conv() local
248 for (int out_channel = 0; out_channel < output_depth; ++out_channel) { in Conv() local
4026 for (int out_channel = 0; out_channel < output_depth; in TransposeConv() local
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Doptimized_ops.h6340 const int out_channel = output_shape.Dims(3); in TransposeConv() local