Searched refs:shape_str (Results 1 – 4 of 4) sorted by relevance
| /OK3568_Linux_fs/external/rknpu2/examples/rknn_yolov5_demo/src/ |
| H A D | main.cc | 42 std::string shape_str = attr->n_dims < 1 ? "" : std::to_string(attr->dims[0]); in dump_tensor_attr() local
44 shape_str += ", " + std::to_string(attr->dims[i]); in dump_tensor_attr()
50 …attr->index, attr->name, attr->n_dims, shape_str.c_str(), attr->n_elems, attr->size, attr->w_strid… in dump_tensor_attr()
|
| /OK3568_Linux_fs/external/rknpu2/examples/rknn_benchmark/src/ |
| H A D | rknn_benchmark.cpp | 74 std::string shape_str = attr->n_dims < 1 ? "" : std::to_string(attr->dims[0]); in dump_tensor_attr() local
76 shape_str += ", " + std::to_string(attr->dims[i]); in dump_tensor_attr()
82 …attr->index, attr->name, attr->n_dims, shape_str.c_str(), attr->n_elems, attr->size, attr->w_strid… in dump_tensor_attr()
|
| /OK3568_Linux_fs/external/rknpu2/examples/rknn_dynamic_shape_input_demo/src/ |
| H A D | rknn_dynshape_inference.cc | 74 std::string shape_str = attr->n_dims < 1 ? "" : std::to_string(attr->dims[0]); in dump_tensor_attr() local
77 shape_str += ", " + std::to_string(attr->dims[i]); in dump_tensor_attr()
83 …attr->index, attr->name, attr->n_dims, shape_str.c_str(), attr->n_elems, attr->size, attr->w_strid… in dump_tensor_attr()
|
| H A D | rknn_dynshape_inference_zero_copy.cc | 74 std::string shape_str = attr->n_dims < 1 ? "" : std::to_string(attr->dims[0]); in dump_tensor_attr() local
77 shape_str += ", " + std::to_string(attr->dims[i]); in dump_tensor_attr()
83 …attr->index, attr->name, attr->n_dims, shape_str.c_str(), attr->n_elems, attr->size, attr->w_strid… in dump_tensor_attr()
|