Searched refs:meta_tensor (Results 1 – 8 of 8) sorted by relevance
| /third_party/mindspore/mindspore/core/ir/ |
| D | meta_tensor.cc | 41 MetaTensor::MetaTensor(const MetaTensor &meta_tensor) in MetaTensor() argument
42 : Value(meta_tensor), data_type_(meta_tensor.data_type()), shape_(meta_tensor.shape()) {} in MetaTensor()
44 MetaTensor &MetaTensor::operator=(const MetaTensor &meta_tensor) { in operator =() argument
45 if (&meta_tensor == this) { in operator =()
49 data_type_ = meta_tensor.data_type(); in operator =()
50 shape_ = meta_tensor.shape(); in operator =()
51 device_info_ = meta_tensor.device_info(); in operator =()
56 bool MetaTensor::operator==(const MetaTensor &meta_tensor) const { in operator ==()
57 return data_type_ == meta_tensor.data_type() && shape_ == meta_tensor.shape(); in operator ==()
|
| D | meta_tensor.h | 78 MetaTensor(const MetaTensor &meta_tensor);
87 virtual MetaTensor &operator=(const MetaTensor &meta_tensor);
95 virtual bool operator==(const MetaTensor &meta_tensor) const;
|
| D | func_graph.cc | 725 ParameterPtr FuncGraph::add_weight(const tensor::MetaTensorPtr &meta_tensor) { in add_weight() argument
727 parameter->set_default_param(MakeValue(meta_tensor)); in add_weight()
728 parameter->set_abstract(meta_tensor->ToAbstract()); in add_weight()
|
| D | func_graph.h | 203 virtual ParameterPtr add_weight(const tensor::MetaTensorPtr &meta_tensor);
|
| /third_party/mindspore/tests/ut/cpp/ir/ |
| D | meta_tensor_test.cc | 46 MetaTensor meta_tensor(TypeId::kNumberTypeFloat64, dimensions); in TEST_F() local
49 ASSERT_EQ(TypeId::kNumberTypeFloat64, meta_tensor.data_type()); in TEST_F()
52 ASSERT_EQ(2, meta_tensor.DimensionSize(0)); in TEST_F()
53 ASSERT_EQ(3, meta_tensor.DimensionSize(1)); in TEST_F()
54 ASSERT_EQ(-1, meta_tensor.DimensionSize(2)); in TEST_F()
57 ASSERT_EQ(6, meta_tensor.ElementsNum()); in TEST_F()
|
| /third_party/mindspore/mindspore/lite/cmake/ |
| D | file_list.cmake | 38 ${CORE_DIR}/ir/meta_tensor.h
|
| /third_party/mindspore/mindspore/ccsrc/pipeline/pynative/ |
| D | pynative_execute.cc | 1005 tensor::MetaTensorPtr meta_tensor = nullptr; in ConstructForwardGraph() local
1007 meta_tensor = obj.cast<tensor::MetaTensorPtr>(); in ConstructForwardGraph()
1008 if (meta_tensor) { in ConstructForwardGraph()
1009 op_mask = meta_tensor->is_parameter(); in ConstructForwardGraph()
1291 auto meta_tensor = obj.cast<tensor::MetaTensorPtr>(); in SetTensorMixPrecisionCast() local
1292 if (meta_tensor && meta_tensor->is_parameter()) { in SetTensorMixPrecisionCast()
|
| /third_party/mindspore/mindspore/lite/tools/dataset/cropper/ |
| D | dependencies.txt | 1 …meta_tensor.cc", "mindspore/ccsrc/minddata/dataset/engine/ir/datasetops/source/tf_record_node.cc",…
|