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 #include <string>
  24
  25 namespace cldnn {
  26 primitive_type_id embed::type_id() {
  27     static primitive_type_base<embed> instance;
  28     return &instance;
  29 }
  30
  31 layout embed_inst::calc_output_layout(embed_node const& node) {
  32     assert(static_cast<bool>(node.get_primitive()->output_data_type) == false &&
  33            "Output data type forcing is not supported for embed_node!");
  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 =
  39         layout(input_layout.data_type,
  40                format::bfyx,
  41                tensor(input_layout.size.batch[0], input_layout.size.spatial[0], weights_layout.size.batch[0], 1));
  42     return result;
  43 }
  44
  45 std::string embed_inst::to_string(embed_node const& node) {
  46     auto desc = node.get_primitive();
  47     auto node_info = node.desc_to_json();
  48     auto bias_id = desc->bias != "" ? desc->bias : "no bias";
  49     auto weights_id = desc->weights;
  50
  51     std::stringstream primitive_description;
  52
  53     json_composite embed_info;
  54     embed_info.add("weights id", weights_id);
  55     embed_info.add("bias id", bias_id);
  56
  57     node_info->add("embed info", embed_info);
  58     node_info->dump(primitive_description);
  59
  60     return primitive_description.str();
  61 }
  62
  63 embed_inst::typed_primitive_inst(network_impl& network, embed_node const& node) : parent(network, node) {
  64     auto input_size = node.input().get_output_layout();
  65     auto output_size = output_memory().get_layout();
  66     CLDNN_ERROR_NOT_PROPER_FORMAT(node.id(),
  67                                   "input format",
  68                                   input_size.format.value,
  69                                   "expected format",
  70                                   format::yxfb,
  71                                   format::bfyx);
  72     CLDNN_ERROR_NOT_EQUAL(node.id(),
  73                           "Input size",
  74                           input_size.size.raw.size(),
  75                           "output size",
  76                           output_size.size.raw.size(),
  77                           "");
  78     CLDNN_ERROR_NOT_EQUAL(node.id(),
  79                           "Input batch",
  80                           input_size.size.batch[0],
  81                           "output batch",
  82                           output_size.size.batch[0],
  83                           "");
  84     CLDNN_ERROR_NOT_EQUAL(node.id(), "Input feature", input_size.size.feature[0], "size 1", 1, "");
  85     CLDNN_ERROR_NOT_EQUAL(node.id(), "Input y size ", input_size.size.spatial[1], "size 1", 1, "");
  86 }
  87 }  // namespace cldnn