Home
last modified time | relevance | path

Searched refs:DE_INT64 (Results 1 – 24 of 24) sorted by relevance

/third_party/mindspore/tests/ut/cpp/dataset/
Dbatch_op_test.cc67 rc = de::Tensor::CreateFromMemory(de::TensorShape({12, 1}), de::DataType(DataType::DE_INT64), in TEST_F()
101 rc = de::Tensor::CreateFromMemory(de::TensorShape({7, 1}), de::DataType(DataType::DE_INT64), in TEST_F()
104 rc = de::Tensor::CreateFromMemory(de::TensorShape({7, 1}), de::DataType(DataType::DE_INT64), in TEST_F()
107 rc = de::Tensor::CreateFromMemory(de::TensorShape({7, 1}), de::DataType(DataType::DE_INT64), in TEST_F()
151 rc = de::Tensor::CreateFromMemory(de::TensorShape({7, 1}), de::DataType(DataType::DE_INT64), in TEST_F()
154 rc = de::Tensor::CreateFromMemory(de::TensorShape({7, 1}), de::DataType(DataType::DE_INT64), in TEST_F()
157 rc = de::Tensor::CreateFromMemory(de::TensorShape({7, 1}), de::DataType(DataType::DE_INT64), in TEST_F()
160 rc = de::Tensor::CreateFromMemory(de::TensorShape({3, 1}), de::DataType(DataType::DE_INT64), in TEST_F()
210 rc = de::Tensor::CreateFromMemory(de::TensorShape({7, 1}), de::DataType(DataType::DE_INT64), in TEST_F()
213 rc = de::Tensor::CreateFromMemory(de::TensorShape({5, 1}), de::DataType(DataType::DE_INT64), in TEST_F()
[all …]
Ddatatype_test.cc61 x = DataType::kTypeInfo[DataType::DE_INT64].sizeInBytes_; in TEST_F()
62 d = DataType(DataType::DE_INT64); in TEST_F()
80 …e::DE_UNKNOWN || d == DataType::DE_UINT32 || d == DataType::DE_UINT64 || d == DataType::DE_INT64) { in FromDT()
118 FromDT(DataType(DataType::DE_INT64), 0, "int64"); in TEST_F()
133 ASSERT_TRUE(DataType(DataType::DE_INT64).IsCompatible<int64_t>()); in TEST_F()
141 ASSERT_FALSE(DataType(DataType::DE_INT64).IsCompatible<double>()); in TEST_F()
Dstand_alone_samplers_test.cc33 RETURN_IF_NOT_OK(Tensor::CreateFromMemory(shape, DataType(DataType::DE_INT64), data, sample_ids)); in CreateINT64Tensor()
56 …Tensor::CreateFromMemory(TensorShape({7}), DataType(DataType::DE_INT64), (unsigned char *)(res[i])… in TEST_F()
Dmap_op_test.cc305 DataType::Type::DE_UINT8, DataType::Type::DE_INT64, in TEST_F()
306 DataType::Type::DE_FLOAT32, DataType::Type::DE_INT64}); in TEST_F()
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/core/
Ddata_type.h48 DE_INT64, enumerator
221 type_ == DataType::DE_INT64; in IsSignedInt()
267 return DataType(DataType::DE_INT64);
337 …return type_ == DataType::DE_INT64 || type_ == DataType::DE_INT32 || type_ == DataType::DE_INT16 ||
Dtype_id.h41 return dataset::DataType(dataset::DataType::DE_INT64); in MSTypeToDEType()
73 case dataset::DataType::DE_INT64: in DETypeToMSType()
Ddata_type.cc108 type_ = DE_INT64; in DataType()
147 return DataType(DataType::DE_INT64); in FromNpArray()
Dtensor.cc409 CASE_PRINT(DataType::DE_INT64, int64_t) in PrintItemAt()
666 } else if (type_ == DataType::DE_INT64) { in to_json()
799 } else if (type_ == DataType::DE_INT64) { in GetDataAsNumpy()
934 case DataType::DE_INT64: { in GetSignedIntAt()
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/gnn/
Dtensor_proto.cc30 …{DataTypePb::DE_PB_INT64, DataType::DE_INT64}, {DataTypePb::DE_PB_UINT64, DataType::DE_UINT64},
40 …{DataType::DE_INT64, DataTypePb::DE_PB_INT64}, {DataType::DE_UINT64, DataTypePb::DE_PB_UINT64},
Dgraph_feature_parser.cc67 …ensor::CreateEmpty(std::move(TensorShape({2})), std::move(DataType(DataType::DE_INT64)), &tensor)); in LoadFeatureToSharedMemory()
Dgraph_data_impl.cc454 RETURN_IF_NOT_OK(Tensor::CreateEmpty(shape, DataType(DataType::DE_INT64), &fea_tensor)); in GetNodeFeatureSharedMemory()
540 RETURN_IF_NOT_OK(Tensor::CreateEmpty(shape, DataType(DataType::DE_INT64), &fea_tensor)); in GetEdgeFeatureSharedMemory()
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/cache/
Dcache_fbb.cc46 CASE(DE_INT64); in SerializeOneTensorMeta()
127 CASE(DE_INT64); in RestoreOneTensor()
Dde_tensor.fbs28 DE_INT64 = 8,
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/kernels/image/
Drgb_to_bgr_op.cc29 … input_type != DataType::DE_INT64 && input_type != DataType::DE_STRING, in Compute()
Dimage_utils.cc834 case DataType::DE_INT64: in Normalize()
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/datasetops/source/
Dalbum_op.cc196 if (data_schema_->Column(col_num).Type() == DataType::DE_INT64) { in LoadIntArrayTensor()
298 if (data_schema_->Column(col_num).Type() == DataType::DE_INT64) { in LoadIntTensor()
393 …(data_schema_->Column(i).Type() == DataType::DE_INT64 || data_schema_->Column(i).Type() == DataTyp… in loadColumnData()
398 …(data_schema_->Column(i).Type() == DataType::DE_INT64 || data_schema_->Column(i).Type() == DataTyp… in loadColumnData()
Dtf_reader_op.cc501 } else if (current_col.Type() == DataType::DE_INT64) { in LoadIntListSwitch()
/third_party/mindspore/mindspore/lite/minddata/wrapper/
Dalbum_op_android.cc280 if (data_schema_->Column(col_num).Type() == DataType::DE_INT64) { in LoadIntArrayTensor()
369 if (data_schema_->Column(col_num).Type() == DataType::DE_INT64) { in LoadIntTensor()
386 …(data_schema_->Column(i).Type() == DataType::DE_INT64 || data_schema_->Column(i).Type() == DataTyp… in LoadIntTensorRowByIndex()
393 …(data_schema_->Column(i).Type() == DataType::DE_INT64 || data_schema_->Column(i).Type() == DataTyp… in LoadIntTensorRowByIndex()
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/kernels/data/
Ddata_utils.cc75 if (input->type() == DataType::DE_INT64) { in OneHotEncodingSigned()
176 case DataType::DE_INT64: { in FillHelper()
280 case DataType::DE_INT64: in CastFrom()
327 case DataType::DE_INT64: in TypeCast()
430 } else if (tensor_type == DataType::DE_INT64) { in PadEndNumeric()
582 case DataType::DE_INT64: in Mask()
819 if (input->type() == DataType::DE_INT64) { in Unique()
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/datasetops/source/sampler/
Dsampler.cc75 …col_desc_ = std::make_unique<ColDescriptor>("sampleIds", DataType(DataType::DE_INT64), TensorImpl:… in CreateSamplerTensor()
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/tdt/
Dtdt_plugin.cc137 case DataType::DE_INT64: in getTdtType()
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/text/kernels/
Dto_number_op.cc45 case DataType::DE_INT64: in Compute()
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/cache/perf/
Dcache_pipeline_run.cc269 …auto col_desc = std::make_unique<ColDescriptor>("int64", DataType(DataType::DE_INT64), TensorImpl:… in WriterWorkerEntry()
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/consumers/
Dtree_consumer.cc503 } else if (column_type == DataType::DE_INT64) { in FetchItemData()