64 class TensorCode (ReferenceCode) :
66 ReferenceCode.__init__(self, ref)
68 def declare_var(self, code_gen) :
69 code_gen.write_line(
'vx_tensor %s;' % self.ref.name)
71 def call_create(self, code_gen) :
72 code_gen.write_if_status();
73 code_gen.write_open_brace();
74 code_gen.write_line(
"vx_size dims[%d];" % (self.ref.number_of_dims))
75 for index, dim
in enumerate(self.ref.dims) :
76 code_gen.write_line(
"dims[%d] = %d;" % (index, dim))
77 if self.ref.access_type ==
"Virtual":
78 code_gen.write_line(
"usecase->%s = vxCreateVirtualTensor(graph, %d, dims, %s, %d);" % (self.ref.name, self.ref.number_of_dims, Type.get_vx_enum_name(self.ref.data_type), self.ref.fixed_point_position));
79 elif self.ref.in_file !=
"./":
80 code_gen.write_line(
'//usecase->%s = create_tensor_from_file(context, "%s", vx_false_e);' % (self.ref.name, self.ref.in_file));
82 code_gen.write_line(
"usecase->%s = vxCreateTensor(context, %d, dims, %s, %d);" % (self.ref.name, self.ref.number_of_dims, Type.get_vx_enum_name(self.ref.data_type), self.ref.fixed_point_position));
83 code_gen.write_line(
"if (usecase->%s == NULL)" % (self.ref.name));
84 code_gen.write_open_brace()
85 code_gen.write_line(
"status = VX_ERROR_NO_RESOURCES;");
86 code_gen.write_close_brace()
87 code_gen.write_if_status();
88 code_gen.write_open_brace();
89 self.set_ref_name(code_gen)
90 code_gen.write_close_brace()
91 code_gen.write_close_brace()
def __init__(self, number_of_dims, dims, data_type, fixed_point_position, access_type="Host", in_file_addr="./", out_file_addr="./", name="default")
Constructor used to create this object.