Skip to content

Commit edd421e

Browse files
committed
fixed warnings
1 parent c2cf6f7 commit edd421e

File tree

2 files changed

+52
-52
lines changed

2 files changed

+52
-52
lines changed

include/cppflow/ops_generator/generator.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -78,7 +78,7 @@ def code(self):
7878
'string' : '''
7979
std::vector<std::size_t> {0}_sizes; {0}_sizes.reserve({0}.size());
8080
std::transform({0}.begin(), {0}.end(), std::back_inserter({0}_sizes), [](const auto& s) {{ return s.size();}});
81-
TFE_OpSetAttrStringList(op, "{orig:}", reinterpret_cast<const void *const *>({0}.data()), {0}_sizes.data(), {0}.size());
81+
TFE_OpSetAttrStringList(op, "{orig:}", reinterpret_cast<const void *const *>({0}.data()), {0}_sizes.data(), (int){0}.size());
8282
''',
8383
'int' : 'TFE_OpSetAttrIntList(op, "{orig:}", {0}.data(), {0}.size());',
8484
'float' : 'TFE_OpSetAttrFloatList(op, "{orig:}", {0}.data(), {0}.size());',
@@ -88,7 +88,7 @@ def code(self):
8888
std::vector<const int64_t*> {0}_values; {0}_values.reserve({0}.size());
8989
std::vector<int> {0}_ndims; {0}_ndims.reserve({0}.size());
9090
std::transform({0}.begin(), {0}.end(), std::back_inserter({0}_values), [](const auto& v) {{ return v.data();}});
91-
std::transform({0}.begin(), {0}.end(), std::back_inserter({0}_ndims), [](const auto& v) {{ return v.size();}});
91+
std::transform({0}.begin(), {0}.end(), std::back_inserter({0}_ndims), [](const auto& v) {{ (int)return v.size();}});
9292
TFE_OpSetAttrShapeList(op, "{orig:}", {0}_values.data(), {0}_ndims.data(), {0}.size(), context::get_status());
9393
status_check(context::get_status());
9494
''',
@@ -97,7 +97,7 @@ def code(self):
9797
else:
9898
return textwrap.dedent({
9999
'shape' : '''
100-
TFE_OpSetAttrShape(op, "{orig:}", {0}.data(), {0}.size(), context::get_status());
100+
TFE_OpSetAttrShape(op, "{orig:}", {0}.data(), (int){0}.size(), context::get_status());
101101
status_check(context::get_status());
102102
''',
103103
'int' : 'TFE_OpSetAttrInt(op, "{orig:}", {0});',
@@ -173,7 +173,7 @@ def code(self):
173173
add_inputs_list = textwrap.dedent('''
174174
std::vector<TFE_TensorHandle*> {0}_handles; {0}_handles.reserve({0}.size());
175175
std::transform({0}.begin(), {0}.end(), std::back_inserter({0}_handles), [](const auto& t) {{ return t.tfe_handle.get();}});
176-
TFE_OpAddInputList(op, {0}_handles.data(), {0}.size(), context::get_status());
176+
TFE_OpAddInputList(op, {0}_handles.data(), (int){0}.size(), context::get_status());
177177
status_check(context::get_status());
178178
''').replace('\n', '\n ')
179179

0 commit comments

Comments
 (0)