Home
last modified time | relevance | path

Searched defs:in_channel (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
Dconv.h73 for (int in_channel = 0; in_channel < input_depth; ++in_channel) { in ConvPerChannel() local
Ddepthwise_conv.h65 for (int in_channel = 0; in_channel < input_depth; ++in_channel) { in DepthwiseConvPerChannel() local
/external/tensorflow/tensorflow/core/kernels/
Ddepthwise_conv_op_gpu.h101 const int in_channel = out_channel / depth_multiplier; in DepthwiseConv2dGPUKernelNHWC() local
350 const int in_channel = out_channel / depth_multiplier; in DepthwiseConv2dGPUKernelNCHW() local
Dconv_ops_using_gemm.cc157 for (int in_channel = 0; in_channel < input_depth; in operator ()() local
Dquantized_conv_ops.cc134 for (int in_channel = 0; in_channel < input_depth; in operator ()() local
Dconv_ops_fused_image_transform.cc526 for (int in_channel = 0; in operator ()() local
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dreference_ops.h174 for (int in_channel = 0; in_channel < input_depth; ++in_channel) { in Conv() local
254 for (int in_channel = 0; in_channel < input_depth; ++in_channel) { in Conv() local
4020 for (int in_channel = 0; in_channel < input_depth; ++in_channel) { in TransposeConv() local
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Doptimized_ops.h6341 const int in_channel = input_shape.Dims(3); in TransposeConv() local