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

   1 /*
   2 // Copyright (c) 2018 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 ///////////////////////////////////////////////////////////////////////////////////////////////////
  19 #include "embed_inst.h"
  20 #include "primitive_type_base.h"
  21 #include "error_handler.h"
  22 #include "json_object.h"
  23
  24 namespace cldnn
  25 {
  26         primitive_type_id embed_type_id()
  27         {
  28                 static primitive_type_base<embed> instance;
  29                 return &instance;
  30         }
  31
  32         layout embed_inst::calc_output_layout(embed_node const& node)
  33         {
  34                 auto input_layout = node.input().get_output_layout();
  35                 auto desc = node.get_primitive();
  36                 auto weights_layout = node.weights().get_output_layout();
  37
  38                 auto result = layout(input_layout.data_type, format::bfyx, tensor(input_layout.size.batch[0], input_layout.size.spatial[0] * input_layout.size.spatial[1], weights_layout.size.batch[0], 1));
  39                 return result;
  40
  41         }
  42
  43         std::string embed_inst::to_string(embed_node const& node)
  44         {
  45                 auto desc = node.get_primitive();
  46                 auto node_info = node.desc_to_json();
  47                 auto bias_id = desc->bias != "" ? desc->bias : "no bias";
  48                 auto weights_id = desc->weights;
  49
  50                 std::stringstream primitive_description;
  51
  52                 json_composite embed_info;
  53                 embed_info.add("weights id", weights_id);
  54                 embed_info.add("bias id", bias_id);
  55
  56                 node_info.add("embed info", embed_info);
  57                 node_info.dump(primitive_description);
  58
  59                 return primitive_description.str();
  60         }
  61
  62         embed_inst::typed_primitive_inst(network_impl& network, embed_node const& node)
  63                 :parent(network, node)
  64         {
  65         auto input_size = node.input().get_output_layout();
  66                 auto output_size = output_memory().get_layout();
  67                 CLDNN_ERROR_NOT_PROPER_FORMAT(node.id(), "input format", input_size.format.value, "expected format", format::yxfb, format::bfyx);
  68                 CLDNN_ERROR_NOT_EQUAL(node.id(), "Input size", input_size.size.raw.size(), "output size", output_size.size.raw.size(), "");
  69         }
  70 }