Cross Reference: /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/depthwiseconv

Lines Matching defs:input_depth
34   static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
81   static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
149   static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
219   static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
296   static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
362   static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
476   static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
536   static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
631   static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
701   static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
786   static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
807       for (; ic <= input_depth - 8; ic += 8) {
857       for (; ic < input_depth; ic++) {
872   static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
882       for (; ic <= input_depth - 8; ic += 8) {
921       for (; ic < input_depth; ic++) {
937   static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
947       for (; ic <= input_depth - 16; ic += 16) {
983       for (; ic <= input_depth - 8; ic += 8) {
1010       for (; ic < input_depth; ic++) {
1022   static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
1076   static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
1110   static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
1154   static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
1210   static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
1259   static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
1291   static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
1354   static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
1418   static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
1469     int stride, int input_depth, int input_width, const uint8* input_data,
1483     TFLITE_DCHECK_EQ(input_depth, kFixedInputDepth);
1488   TFLITE_DCHECK_EQ(output_depth, input_depth * depth_multiplier);
1489   const int input_ptr_increment = stride * input_depth;
1525     const uint8* input_ptr = input_data + in_x_origin * input_depth;
1529         kFixedDepthMultiplier>::Run(num_output_pixels, input_depth,
1539     int stride, int input_depth, int input_width, const uint8* input_data,
1562       << "* input_depth = " << input_depth << "\n"
1581     const uint8* input_ptr = input_data + in_x_origin * input_depth;
1582     const int input_ptr_increment = (stride - 1) * input_depth;
1585       for (int ic = 0; ic < input_depth; ++ic) {
1688   const int input_depth = ArraySize(input_dims, 0);
1693   TFLITE_DCHECK(output_depth == input_depth * depth_multiplier);
1713       (input_depth == FIXED_INPUT_DEPTH || FIXED_INPUT_DEPTH == 0) &&     \
1790               stride_width, input_depth, input_width,