Changes to generator file to produce static cast of size_t values being passed into TensorFlow int parameters.

Author: 8-byte

include/cppflow/ops_generator/generator.py

@@ -78,26 +78,26 @@ def code(self):
                 'string' : '''
                             std::vector<std::size_t> {0}_sizes; {0}_sizes.reserve({0}.size());
                             std::transform({0}.begin(), {0}.end(), std::back_inserter({0}_sizes), [](const auto& s) {{ return s.size();}});
-                            TFE_OpSetAttrStringList(op.get(), "{orig:}", reinterpret_cast<const void *const *>({0}.data()), {0}_sizes.data(), {0}.size());
+                            TFE_OpSetAttrStringList(op.get(), "{orig:}", reinterpret_cast<const void *const *>({0}.data()), {0}_sizes.data(), static_cast<int>({0}.size()));
                             ''',
-                'int'    : 'TFE_OpSetAttrIntList(op.get(), "{orig:}", {0}.data(), {0}.size());',
-                'float'  : 'TFE_OpSetAttrFloatList(op.get(), "{orig:}", {0}.data(), {0}.size());',
+                'int'    : 'TFE_OpSetAttrIntList(op.get(), "{orig:}", {0}.data(), static_cast<int>({0}.size()));',
+                'float'  : 'TFE_OpSetAttrFloatList(op.get(), "{orig:}", {0}.data(), static_cast<int>({0}.size()));',
                 'bool'   : 'TFE_OpSetAttrBoolList(op.get(), "{orig:}", std::vector<unsigned char>({0}.begin(), {0}.end()).data(), {0}.size());',
-                'type'   : 'TFE_OpSetAttrTypeList(op.get(), "{orig:}", reinterpret_cast<const enum TF_DataType *>({0}.data()), {0}.size());',
+                'type'   : 'TFE_OpSetAttrTypeList(op.get(), "{orig:}", reinterpret_cast<const enum TF_DataType *>({0}.data()), static_cast<int>({0}.size()));',
                 'shape'  : '''
                             std::vector<const int64_t*> {0}_values; {0}_values.reserve({0}.size());
                             std::vector<int> {0}_ndims; {0}_ndims.reserve({0}.size());
                             std::transform({0}.begin(), {0}.end(), std::back_inserter({0}_values), [](const auto& v) {{ return v.data();}});
-                            std::transform({0}.begin(), {0}.end(), std::back_inserter({0}_ndims), [](const auto& v) {{ return v.size();}});
-                            TFE_OpSetAttrShapeList(op.get(), "{orig:}", {0}_values.data(), {0}_ndims.data(), {0}.size(), context::get_status());
+                            std::transform({0}.begin(), {0}.end(), std::back_inserter({0}_ndims), [](const auto& v) {{ return static_cast<int>(v.size());}});
+                            TFE_OpSetAttrShapeList(op.get(), "{orig:}", {0}_values.data(), {0}_ndims.data(), static_cast<int>({0}.size()), context::get_status());
                             status_check(context::get_status());
                             ''',
             }[self.type].format(self.name.replace('template', 'template_arg'), orig=self.name)).replace('\n', '\n    ')
 
         else:
             return textwrap.dedent({
                 'shape' : '''
-                          TFE_OpSetAttrShape(op.get(), "{orig:}", {0}.data(), {0}.size(), context::get_status());
+                          TFE_OpSetAttrShape(op.get(), "{orig:}", {0}.data(), static_cast<int>({0}.size()), context::get_status());
                           status_check(context::get_status());
                            ''',
                 'int'   : 'TFE_OpSetAttrInt(op.get(), "{orig:}", {0});',
@@ -172,7 +172,7 @@ def code(self):
         add_inputs_list = textwrap.dedent('''
             std::vector<TFE_TensorHandle*> {0}_handles; {0}_handles.reserve({0}.size());
             std::transform({0}.begin(), {0}.end(), std::back_inserter({0}_handles), [](const auto& t) {{ return t.tfe_handle.get();}});
-            TFE_OpAddInputList(op.get(), {0}_handles.data(), {0}.size(), context::get_status());
+            TFE_OpAddInputList(op.get(), {0}_handles.data(), static_cast<int>({0}.size()), context::get_status());
             status_check(context::get_status());
         ''').replace('\n', '\n    ')
 

include/cppflow/raw_ops.h

@@ -4127,7 +4127,7 @@ inline tensor choose_fastest_dataset(const std::vector<tensor>&input_datasets, i
 
 
     // Attributes
-    TFE_OpSetAttrInt(op.get(), "N", static_cast<int>(input_datasets.size()));
+    TFE_OpSetAttrInt(op.get(), "N", input_datasets.size());
     TFE_OpSetAttrInt(op.get(), "num_experiments", num_experiments);
     TFE_OpSetAttrTypeList(op.get(), "output_types", reinterpret_cast<const enum TF_DataType *>(output_types.data()), static_cast<int>(output_types.size()));
 
@@ -6132,6 +6132,32 @@ inline tensor decode_gif(const tensor& contents) {
 }
 
 
+inline tensor decode_image(const tensor& contents, int64_t channels=0, datatype dtype=static_cast<datatype>(4), bool expand_animations=true) {
+
+    // Define Op
+    std::unique_ptr<TFE_Op, decltype(&TFE_DeleteOp)> op(TFE_NewOp(context::get_context(), "DecodeImage", context::get_status()), &TFE_DeleteOp);
+    status_check(context::get_status());
+
+    // Required input arguments
+    
+    TFE_OpAddInput(op.get(), contents.tfe_handle.get(), context::get_status());
+    status_check(context::get_status());
+    
+
+    // Attributes
+    TFE_OpSetAttrInt(op.get(), "channels", channels);
+    TFE_OpSetAttrType(op.get(), "dtype", dtype);
+    TFE_OpSetAttrBool(op.get(), "expand_animations", (unsigned char)expand_animations);
+
+    // Execute Op
+    int num_outputs_op = 1;
+    TFE_TensorHandle* res[1] = {nullptr};
+    TFE_Execute(op.get(), res, &num_outputs_op, context::get_status());
+    status_check(context::get_status());
+    return tensor(res[0]);
+}
+
+
 inline tensor decode_j_s_o_n_example(const tensor& json_examples) {
 
     // Define Op