IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Segment Sum renamed in common lib (#17378)

Browse Source
This commit is contained in:
Evgenya Stepyreva 2023-05-08 11:00:58 +04:00 committed by GitHub
parent 94a7342548
commit af3d1d69dd
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/frontends/tensorflow_lite/src/op/space_to_depth.cpp
View File

@ -16,7 +16,7 @@ namespace op {
OutputVector space_to_depth(const ov::frontend::tensorflow_lite::NodeContext& node) {
const auto& decoder = get_decoder(node);
std::map<std::string, ov::Any> attrs{
{"seed", static_cast<int64_t>(decoder->get_attribute(&tflite::SpaceToDepthOptions::block_size))},
{"block_size", static_cast<int64_t>(decoder->get_attribute(&tflite::SpaceToDepthOptions::block_size))},
};
return attribute_helper(node, attrs, ov::frontend::tensorflow::op::translate_space_to_depth_op, "SpaceToDepth");
}

2
src/frontends/tensorflow_lite/src/op_table.cpp
View File

@ -156,7 +156,7 @@ std::map<std::string, CreatorFunction> get_supported_ops() {
{"ROUND", DEQUANTIZE_INPUTS(translate_round_op)},
{"RSQRT", DEQUANTIZE_INPUTS(translate_rsqrt_op)},
{"SCATTER_ND", DEQUANTIZE_INPUTS(translate_scatter_nd_op)},
{"SEGMENT_SUM", DEQUANTIZE_INPUTS(translate_segment_sum_op)},
{"SEGMENT_SUM", OP_CONVERT_TYPE_RENAME(translate_segment_sum_op, "SegmentSum")},
{"SELECT", OP_CONVERT_TYPE_RENAME(translate_select_op, "Select")},
{"SELECT_V2", OP_CONVERT_TYPE_RENAME(translate_select_v2_op, "SelectV2")},
{"SHAPE", shape},