Home
last modified time | relevance | path

Searched refs:TensorImpl (Results 1 – 25 of 39) sorted by relevance

12

/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/ir/datasetops/source/
Dcoco_node.cc96 …schema->AddColumn(ColDescriptor(std::string("image"), DataType(DataType::DE_UINT8), TensorImpl::kF… in Build()
100 … ColDescriptor(std::string("bbox"), DataType(DataType::DE_FLOAT32), TensorImpl::kFlexible, 1))); in Build()
102 …ColDescriptor(std::string("category_id"), DataType(DataType::DE_UINT32), TensorImpl::kFlexible, 1)… in Build()
104 … ColDescriptor(std::string("iscrowd"), DataType(DataType::DE_UINT32), TensorImpl::kFlexible, 1))); in Build()
108 …ColDescriptor(std::string("segmentation"), DataType(DataType::DE_FLOAT32), TensorImpl::kFlexible, … in Build()
110 … ColDescriptor(std::string("iscrowd"), DataType(DataType::DE_UINT32), TensorImpl::kFlexible, 1))); in Build()
114 …ColDescriptor(std::string("keypoints"), DataType(DataType::DE_FLOAT32), TensorImpl::kFlexible, 1))… in Build()
116 …ColDescriptor(std::string("num_keypoints"), DataType(DataType::DE_UINT32), TensorImpl::kFlexible, … in Build()
120 … ColDescriptor(std::string("bbox"), DataType(DataType::DE_FLOAT32), TensorImpl::kFlexible, 1))); in Build()
122 …ColDescriptor(std::string("category_id"), DataType(DataType::DE_UINT32), TensorImpl::kFlexible, 1)… in Build()
[all …]
Dvoc_node.cc100 …ColDescriptor(std::string(kColumnImage), DataType(DataType::DE_UINT8), TensorImpl::kFlexible, 1))); in Build()
102 …ColDescriptor(std::string(kColumnTarget), DataType(DataType::DE_UINT8), TensorImpl::kFlexible, 1))… in Build()
106 …ColDescriptor(std::string(kColumnImage), DataType(DataType::DE_UINT8), TensorImpl::kFlexible, 1))); in Build()
108 …ColDescriptor(std::string(kColumnBbox), DataType(DataType::DE_FLOAT32), TensorImpl::kFlexible, 1))… in Build()
110 …ColDescriptor(std::string(kColumnLabel), DataType(DataType::DE_UINT32), TensorImpl::kFlexible, 1))… in Build()
112 …ColDescriptor(std::string(kColumnDifficult), DataType(DataType::DE_UINT32), TensorImpl::kFlexible,… in Build()
114 …ColDescriptor(std::string(kColumnTruncate), DataType(DataType::DE_UINT32), TensorImpl::kFlexible, … in Build()
120 … ColDescriptor(meta_file_name, DataType(DataType::DE_STRING), TensorImpl::kFlexible, 0, &scalar))); in Build()
Dcityscapes_node.cc77 …OT_OK(schema->AddColumn(ColDescriptor("image", DataType(DataType::DE_UINT8), TensorImpl::kCv, 1))); in Build()
81 …schema->AddColumn(ColDescriptor("task", DataType(DataType::DE_STRING), TensorImpl::kFlexible, 0, &… in Build()
84 …schema->AddColumn(ColDescriptor("task", DataType(DataType::DE_UINT8), TensorImpl::kFlexible, 0, &s… in Build()
Dcifar100_node.cc63 …OT_OK(schema->AddColumn(ColDescriptor("image", DataType(DataType::DE_UINT8), TensorImpl::kCv, 1))); in Build()
66 …schema->AddColumn(ColDescriptor("coarse_label", DataType(DataType::DE_UINT32), TensorImpl::kFlexib… in Build()
68 …schema->AddColumn(ColDescriptor("fine_label", DataType(DataType::DE_UINT32), TensorImpl::kFlexible… in Build()
Dsbu_node.cc65 …schema->AddColumn(ColDescriptor("image", DataType(DataType::DE_UINT8), TensorImpl::kFlexible, 1))); in Build()
66 …hema->AddColumn(ColDescriptor("caption", DataType(DataType::DE_UINT8), TensorImpl::kFlexible, 1))); in Build()
Dcifar10_node.cc63 …OT_OK(schema->AddColumn(ColDescriptor("image", DataType(DataType::DE_UINT8), TensorImpl::kCv, 1))); in Build()
66 …schema->AddColumn(ColDescriptor("label", DataType(DataType::DE_UINT32), TensorImpl::kFlexible, 0, … in Build()
Dusps_node.cc77 …OT_OK(schema->AddColumn(ColDescriptor("image", DataType(DataType::DE_UINT8), TensorImpl::kCv, 1))); in Build()
80 …schema->AddColumn(ColDescriptor("label", DataType(DataType::DE_UINT32), TensorImpl::kFlexible, 0, … in Build()
Ddiv2k_node.cc94 …OK(schema->AddColumn(ColDescriptor("hr_image", DataType(DataType::DE_UINT8), TensorImpl::kCv, 1))); in Build()
97 …schema->AddColumn(ColDescriptor("lr_image", DataType(DataType::DE_UINT8), TensorImpl::kFlexible, 0… in Build()
Dmnist_node.cc59 …OT_OK(schema->AddColumn(ColDescriptor("image", DataType(DataType::DE_UINT8), TensorImpl::kCv, 1))); in Build()
62 …schema->AddColumn(ColDescriptor("label", DataType(DataType::DE_UINT32), TensorImpl::kFlexible, 0, … in Build()
Dflickr_node.cc94 …OT_OK(schema->AddColumn(ColDescriptor("image", DataType(DataType::DE_UINT8), TensorImpl::kCv, 1))); in Build()
97 …schema->AddColumn(ColDescriptor("annotation", DataType(DataType::DE_STRING), TensorImpl::kFlexible… in Build()
Dimage_folder_node.cc72 …schema->AddColumn(ColDescriptor("image", DataType(DataType::DE_UINT8), TensorImpl::kFlexible, 1))); in Build()
74 …schema->AddColumn(ColDescriptor("label", DataType(DataType::DE_INT32), TensorImpl::kFlexible, 0, &… in Build()
Dmanifest_node.cc96 …OT_OK(schema->AddColumn(ColDescriptor("image", DataType(DataType::DE_UINT8), TensorImpl::kCv, 1))); in Build()
99 …schema->AddColumn(ColDescriptor("label", DataType(DataType::DE_UINT32), TensorImpl::kFlexible, 0, … in Build()
Dceleba_node.cc70 …schema->AddColumn(ColDescriptor("image", DataType(DataType::DE_UINT8), TensorImpl::kFlexible, 1))); in Build()
72 …schema->AddColumn(ColDescriptor("attr", DataType(DataType::DE_UINT32), TensorImpl::kFlexible, 1))); in Build()
/third_party/mindspore/tests/ut/cpp/dataset/
Dtensorshape_test.cc116 ColDescriptor c1("col1", DataType(DataType::DE_INT8), TensorImpl::kFlexible, rank); in TEST_F()
126 …ColDescriptor c2("col2", DataType(DataType::DE_INT8), TensorImpl::kFlexible, rank, &requested_shap… in TEST_F()
136 …ColDescriptor c3("col3", DataType(DataType::DE_INT8), TensorImpl::kFlexible, rank, &requested_shap… in TEST_F()
146 …ColDescriptor c4("col4", DataType(DataType::DE_INT8), TensorImpl::kFlexible, rank, &requested_shap… in TEST_F()
156 …ColDescriptor c5("col5", DataType(DataType::DE_INT8), TensorImpl::kFlexible, rank, &requested_shap… in TEST_F()
166 …ColDescriptor c6("col6", DataType(DataType::DE_INT8), TensorImpl::kFlexible, rank, &requested_shap… in TEST_F()
174 …ColDescriptor c7("col7", DataType(DataType::DE_INT8), TensorImpl::kFlexible, rank, &requested_shap… in TEST_F()
Dcifar_op_test.cc58 …(void)schema->AddColumn(ColDescriptor("image", DataType(DataType::DE_UINT8), TensorImpl::kFlexible… in Cifarop()
60 …(void)schema->AddColumn(ColDescriptor("label", DataType(DataType::DE_UINT32), TensorImpl::kFlexibl… in Cifarop()
63 … ColDescriptor("coarse_label", DataType(DataType::DE_UINT32), TensorImpl::kFlexible, 0, &scalar)); in Cifarop()
66 …ColDescriptor("fine_label", DataType(DataType::DE_UINT32), TensorImpl::kFlexible, 0, &another_scal… in Cifarop()
Dcache_op_test.cc86 ColDescriptor c1("image", DataType(DataType::DE_INT8), TensorImpl::kFlexible, in TEST_F()
91 ColDescriptor c2("label", DataType(DataType::DE_UINT32), TensorImpl::kFlexible, rank, &c2Shape); in TEST_F()
244 ColDescriptor c1("image", DataType(DataType::DE_INT8), TensorImpl::kFlexible, in TEST_F()
250 ColDescriptor c2("label", DataType(DataType::DE_UINT32), TensorImpl::kFlexible, rank, &c2Shape); in TEST_F()
360 ColDescriptor c1("image", DataType(DataType::DE_INT8), TensorImpl::kFlexible, in TEST_F()
366 ColDescriptor c2("label", DataType(DataType::DE_UINT32), TensorImpl::kFlexible, rank, &c2Shape); in TEST_F()
469 …rc = schema->AddColumn(ColDescriptor("image", DataType(DataType::DE_UINT8), TensorImpl::kFlexible,… in TEST_F()
471 …rc = schema->AddColumn(ColDescriptor("label", DataType(DataType::DE_INT32), TensorImpl::kFlexible,… in TEST_F()
Dceleba_op_test.cc48 …(void)schema->AddColumn(ColDescriptor("image", DataType(DataType::DE_UINT8), TensorImpl::kFlexible… in Celeba()
49 …(void)schema->AddColumn(ColDescriptor("attr", DataType(DataType::DE_UINT32), TensorImpl::kFlexible… in Celeba()
Dmanifest_op_test.cc49 schema->AddColumn(ColDescriptor("image", DataType(DataType::DE_UINT8), TensorImpl::kFlexible, 1));
50 …schema->AddColumn(ColDescriptor("label", DataType(DataType::DE_UINT32), TensorImpl::kFlexible, 1));
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/
Ddata_schema.cc37 out_type = TensorImpl::kCv; \
39 out_type = TensorImpl::kFlexible; \
41 out_type = TensorImpl::kNP; \
43 out_type = TensorImpl::kNone; \
49 …: type_(DataType::DE_UNKNOWN), rank_(0), tensor_impl_(TensorImpl::kNone), tensor_shape_(nullptr) {} in ColDescriptor()
52 ColDescriptor::ColDescriptor(const std::string &col_name, DataType col_type, TensorImpl tensor_impl… in ColDescriptor()
276 TensorImpl t_impl_value = TensorImpl::kFlexible; in ColumnLoad()
Ddata_schema.h46 …ColDescriptor(const std::string &col_name, DataType col_type, TensorImpl tensor_impl, int32_t rank,
104 TensorImpl GetTensorImpl() const { return tensor_impl_; } in GetTensorImpl()
109 TensorImpl tensor_impl_; // The initial flavour of the tensor for this column
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/datasetops/source/
Dmindrecord_op.cc88 col_desc = ColDescriptor(colname, t_dtype, TensorImpl::kFlexible, 1); in Init()
90 col_desc = ColDescriptor(colname, t_dtype, TensorImpl::kFlexible, 0); in Init()
95 col_desc = ColDescriptor(colname, t_dtype, TensorImpl::kFlexible, t_shape.Rank(), &t_shape); in Init()
98 col_desc = ColDescriptor(colname, t_dtype, TensorImpl::kFlexible, t_shape.Rank(), &t_shape); in Init()
Dcifar_op.cc298 …new_schema->AddColumn(ColDescriptor("image", DataType(DataType::DE_UINT8), TensorImpl::kFlexible, … in CountTotalRows()
301 …new_schema->AddColumn(ColDescriptor("label", DataType(DataType::DE_UINT32), TensorImpl::kFlexible,… in CountTotalRows()
304 … ColDescriptor("coarse_label", DataType(DataType::DE_UINT32), TensorImpl::kFlexible, 0, &scalar))); in CountTotalRows()
307 …ColDescriptor("fine_label", DataType(DataType::DE_UINT32), TensorImpl::kFlexible, 0, &another_scal… in CountTotalRows()
Dcityscapes_op.cc251 …K(new_schema->AddColumn(ColDescriptor("image", DataType(DataType::DE_UINT8), TensorImpl::kCv, 1))); in CountTotalRows()
255 …new_schema->AddColumn(ColDescriptor("task", DataType(DataType::DE_STRING), TensorImpl::kFlexible, … in CountTotalRows()
258 …new_schema->AddColumn(ColDescriptor("task", DataType(DataType::DE_UINT8), TensorImpl::kFlexible, 0… in CountTotalRows()
Dflickr_op.cc225 …new_schema->AddColumn(ColDescriptor("image", DataType(DataType::DE_UINT8), TensorImpl::kFlexible, … in CountTotalRows()
227 ColDescriptor("annotation", DataType(DataType::DE_STRING), TensorImpl::kFlexible, 0, &scalar))); in CountTotalRows()
Dsbu_op.cc105 …schema->AddColumn(ColDescriptor("image", DataType(DataType::DE_UINT8), TensorImpl::kFlexible, 1))); in CountTotalRows()
106 …hema->AddColumn(ColDescriptor("caption", DataType(DataType::DE_UINT8), TensorImpl::kFlexible, 1))); in CountTotalRows()

12