layout crop_inst::calc_output_layout(crop_node const& node)
{
- auto input_layout = node.input().get_output_layout();
- auto result = layout({ input_layout.data_type, input_layout.format, node.get_primitive()->reference_input });
- return result;
+ assert((bool)node.get_primitive()->output_data_type == false
+ && "Output data type forcing is not supported for crop_node!");
+ const auto& ref_in_sizes = node.get_primitive()->reference_input;
+ const auto in_layout = node.input().get_output_layout();
+ const auto& in_sizes = in_layout.size;
+ const auto& offsets = node.get_primitive()->offsets;
+
+ // Check for borders variant of crop.
+ if (ref_in_sizes.batch[0] < 0 || ref_in_sizes.feature[0] < 0 ||
+ ref_in_sizes.spatial[0] < 0 || ref_in_sizes.spatial[1] < 0)
+ {
+ // Ignore not supported dimensions.
+ const auto rb_sizes = ref_in_sizes.negate().sub({0, 0, 0, 0});
+ const auto lt_sizes = offsets.sub({0, 0, 0, 0});
+
+ const auto out_sizes = in_sizes - (rb_sizes + lt_sizes);
+
+ return layout({in_layout.data_type, in_layout.format, out_sizes});
+ }
+ return layout({in_layout.data_type, in_layout.format, ref_in_sizes});
}
std::string crop_inst::to_string(crop_node const& node)
{
- auto desc = node.get_primitive();
- auto offsets = desc->offsets;
+ const auto& desc = node.get_primitive();
+ auto ref_in_sizes = desc->reference_input;
+ const auto& offsets = desc->offsets;
+ const auto in_layout = node.input().get_output_layout();
+ const auto& in_sizes = in_layout.size;
+
auto node_info = node.desc_to_json();
- auto ref_input = desc->reference_input;
-
+
+ // Check for borders variant of crop.
+ if (ref_in_sizes.batch[0] < 0 || ref_in_sizes.feature[0] < 0 ||
+ ref_in_sizes.spatial[0] < 0 || ref_in_sizes.spatial[1] < 0)
+ {
+ // Ignore not supported dimensions.
+ const auto rb_sizes = ref_in_sizes.negate().sub({0, 0, 0, 0});
+ const auto lt_sizes = offsets.sub({0, 0, 0, 0});
+
+ ref_in_sizes = in_sizes - (rb_sizes + lt_sizes);
+ }
+
std::stringstream primitive_description;
json_composite crop_info;
- crop_info.add("reference input", ref_input.to_string());
- crop_info.add("offset", offsets.to_string());
+ crop_info.add("reference input size", ref_in_sizes.to_string());
+ crop_info.add("offset", offsets.to_string());
node_info->add("crop info", crop_info);
node_info->dump(primitive_description);
}
crop_inst::typed_primitive_inst(network_impl& network, crop_node const& node)
- :parent(network, node)
+ : parent(network, node)
{
- auto reference_input_sizes = argument.reference_input;
- auto inp_layout = node.input().get_output_layout();
- auto input_sizes = inp_layout.size;
- auto input_format = inp_layout.format;
- auto offsets = argument.offsets;
+ const auto& ref_in_sizes = argument.reference_input;
+ const auto in_layout = node.input().get_output_layout();
+ const auto& in_sizes = in_layout.size;
+ const auto in_format = in_layout.format;
+ const auto& offsets = argument.offsets;
+
+ CLDNN_ERROR_NOT_PROPER_FORMAT(node.id(), "Input format", in_format.value, "supported crop input formats", format::yxfb, format::bfyx, format::fyxb);
+
+ // Check for borders variant of crop.
+ if (ref_in_sizes.batch[0] < 0 || ref_in_sizes.feature[0] < 0 ||
+ ref_in_sizes.spatial[0] < 0 || ref_in_sizes.spatial[1] < 0)
+ {
+ // Ignore not supported dimensions.
+ const auto rb_sizes = ref_in_sizes.negate().sub({0, 0, 0, 0});
+ const auto lt_sizes = offsets.sub({0, 0, 0, 0});
+
+ const auto out_sizes = in_sizes - (rb_sizes + lt_sizes);
- CLDNN_ERROR_NOT_PROPER_FORMAT(node.id(), "Input format", input_format.value, "supported crop input formats", format::yxfb, format::bfyx );
+ CLDNN_ERROR_TENSOR_SIZES_LESS_THAN(node.id(), "Left/top/lower borders", lt_sizes, "0 value", {}, "Invalid border size: negative");
+ CLDNN_ERROR_TENSOR_SIZES_LESS_THAN(node.id(), "Right/bottom/upper borders", rb_sizes, "0 value", {}, "Invalid border size: negative");
+
+ CLDNN_ERROR_TENSOR_SIZES_LESS_THAN(node.id(), "Input sizes - border sizes", out_sizes, "1 value", {1, 1, 1, 1}, "Invalid border sizes: greater-equal input sizes");
+ }
//check if output sizes matches reference input sizes
- CLDNN_ERROR_TENSOR_SIZES_GREATER_THAN(node.id(), "Reference input", reference_input_sizes, "input sizes", input_sizes, "Reference input tensor/ input tensor mismtach");
-
+ CLDNN_ERROR_TENSOR_SIZES_GREATER_THAN(node.id(), "Reference input", ref_in_sizes, "input sizes", in_sizes, "Reference input tensor/ input tensor mismtach");
+
//check if offsets do not extend input sizes and if match the output sizes
- CLDNN_ERROR_TENSOR_SIZES_LESS_THAN(node.id(), "Batch offsets", offsets, "0 value", { 0, 0, 0, 0 }, "Invalid Batch offset: negative value");
- auto input_size_sub_offsets = input_sizes - offsets;
- CLDNN_ERROR_TENSOR_SIZES_LESS_THAN(node.id(), "input sizes - offsets", input_size_sub_offsets, "reference input sizes", reference_input_sizes, "Invalid Batch offset: exceeds data for output!");
+ CLDNN_ERROR_TENSOR_SIZES_LESS_THAN(node.id(), "Batch offsets", offsets, "0 value", {}, "Invalid Batch offset: negative value");
+ auto input_size_sub_offsets = in_sizes - offsets;
+ CLDNN_ERROR_TENSOR_SIZES_LESS_THAN(node.id(), "input sizes - offsets", input_size_sub_offsets, "reference input sizes", ref_in_sizes, "Invalid Batch offset: exceeds data for output!");
if (node.can_be_optimized())
{
{
_output = _network.get_engine().reinterpret_buffer(input_memory(), node.get_output_layout());
}
-}
\ No newline at end of file
+}
projects / platform / upstream / dldt.git / blobdiff