inference-engine/thirdparty/clDNN/src/fully_connected.cpp

   1 /*
   2 // Copyright (c) 2016-2019 Intel Corporation
   3 //
   4 // Licensed under the Apache License, Version 2.0 (the "License");
   5 // you may not use this file except in compliance with the License.
   6 // You may obtain a copy of the License at
   7 //
   8 //      http://www.apache.org/licenses/LICENSE-2.0
   9 //
  10 // Unless required by applicable law or agreed to in writing, software
  11 // distributed under the License is distributed on an "AS IS" BASIS,
  12 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13 // See the License for the specific language governing permissions and
  14 // limitations under the License.
  15 */
  16
  17 ///////////////////////////////////////////////////////////////////////////////////////////////////
  18 #include "fully_connected_inst.h"
  19 #include "primitive_type_base.h"
  20 #include "error_handler.h"
  21 #include "json_object.h"
  22
  23 namespace cldnn
  24 {
  25 primitive_type_id fully_connected_type_id()
  26 {
  27     static primitive_type_base<fully_connected> instance;
  28     return &instance;
  29 }
  30
  31 namespace
  32 {
  33 bool is_batch_after_spatial(const std::string order)
  34 {
  35     bool spatial_found = false;
  36     for (auto c : order)
  37     {
  38         switch (c)
  39         {
  40         case 'b':
  41         case 'n':
  42             return spatial_found;
  43
  44         case 'x':
  45         case 'y':
  46         case 'z':
  47         case 'w':
  48         case 's':
  49             spatial_found = true;
  50             break;
  51
  52         default: break;
  53         }
  54     }
  55     return false;
  56 }
  57 }
  58
  59 layout fully_connected_inst::calc_output_layout(fully_connected_node const& node)
  60 {
  61     assert((bool)node.get_primitive()->output_data_type == false
  62            && "Output data type forcing is not supported for "
  63               "fully_connected_node!");
  64     auto desc = node.get_primitive();
  65
  66     auto input_layout = node.input().get_output_layout();
  67     auto weights_layout = node.weights().get_output_layout();
  68
  69     if(is_batch_after_spatial(input_layout.format.order()) ||
  70         (input_layout.format == format::bfyx &&                //this condition tests whether our input is batch>1 in bfyx format, if yes there will be
  71         input_layout.size.batch[0] > 1))                            //extra reorder between input and this fc from bfyx to yxfb format (so "is_batch_after_spatial" should return true)
  72     {
  73         auto result = layout(input_layout.data_type, format::yxfb, tensor(input_layout.size.batch[0], weights_layout.size.batch[0], 1, 1));
  74         return result;
  75     }
  76     else
  77     {
  78         auto result = layout(input_layout.data_type, format::bfyx, tensor(input_layout.size.batch[0], weights_layout.size.batch[0], 1, 1));
  79         return result;
  80     }
  81 }
  82
  83 std::string fully_connected_inst::to_string(fully_connected_node const& node)
  84 {
  85     auto desc       = node.get_primitive();
  86     auto node_info  = node.desc_to_json();
  87     auto bias_id    = desc->bias != "" ? desc->bias : "no bias";
  88     auto weights_id = desc->weights;
  89     auto activation = desc->with_activation ? " true" : "false";
  90
  91     std::stringstream primitive_description;
  92
  93     json_composite fc_info;
  94     fc_info.add("weights id", weights_id);
  95     fc_info.add("bias id", bias_id);
  96     fc_info.add("with activation", activation);
  97
  98     node_info->add("fully connected info", fc_info);
  99     node_info->dump(primitive_description);
 100
 101     return primitive_description.str();
 102 }
 103
 104 fully_connected_inst::typed_primitive_inst(network_impl& network, fully_connected_node const& node)
 105     :parent(network, node)
 106 {
 107     auto input_layout = node.input().get_output_layout();
 108     auto output_layout = node.get_output_layout();
 109
 110     CLDNN_ERROR_NOT_PROPER_FORMAT(node.id(), "input format", input_layout.format.value, "expected format", format::yxfb, format::bfyx, format::byxf_af32, format::fs_bs_yx_bsv4_fsv32, format::b_fs_yx_fsv4);
 111     CLDNN_ERROR_NOT_EQUAL(node.id(), "Input size", input_layout.size.raw.size(), "output size", output_layout.size.raw.size(), "");
 112 }
 113 }