@@ -78,7 +78,7 @@ def code(self):
78
78
'string' : '''
79
79
std::vector<std::size_t> {0}_sizes; {0}_sizes.reserve({0}.size());
80
80
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());
82
82
''' ,
83
83
'int' : 'TFE_OpSetAttrIntList(op, "{orig:}", {0}.data(), {0}.size());' ,
84
84
'float' : 'TFE_OpSetAttrFloatList(op, "{orig:}", {0}.data(), {0}.size());' ,
@@ -88,7 +88,7 @@ def code(self):
88
88
std::vector<const int64_t*> {0}_values; {0}_values.reserve({0}.size());
89
89
std::vector<int> {0}_ndims; {0}_ndims.reserve({0}.size());
90
90
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();}});
92
92
TFE_OpSetAttrShapeList(op, "{orig:}", {0}_values.data(), {0}_ndims.data(), {0}.size(), context::get_status());
93
93
status_check(context::get_status());
94
94
''' ,
@@ -97,7 +97,7 @@ def code(self):
97
97
else :
98
98
return textwrap .dedent ({
99
99
'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());
101
101
status_check(context::get_status());
102
102
''' ,
103
103
'int' : 'TFE_OpSetAttrInt(op, "{orig:}", {0});' ,
@@ -173,7 +173,7 @@ def code(self):
173
173
add_inputs_list = textwrap .dedent ('''
174
174
std::vector<TFE_TensorHandle*> {0}_handles; {0}_handles.reserve({0}.size());
175
175
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());
177
177
status_check(context::get_status());
178
178
''' ).replace ('\n ' , '\n ' )
179
179
0 commit comments