Publishing 2019 R3 content
[platform/upstream/dldt.git] / inference-engine / thirdparty / clDNN / src / one_hot.cpp
index c779bea..9345b35 100644 (file)
 #include <vector>
 
 namespace cldnn {
-primitive_type_id one_hot_type_id() {
+primitive_type_id one_hot::type_id() {
     static primitive_type_base<one_hot> instance;
     return &instance;
 }
 
-static bool is_output_bfzyx(layout input, int32_t axis) {
+static bool is_output_bfzyx(const layout& input, int32_t axis) {
     if (input.format == format::bfzyx)
         return true;
     if (axis == 4)