Home
last modified time | relevance | path

Searched defs:tensor_abs (Results 1 – 12 of 12) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/cxx_api/model/acl/
Dacl_model_multi.cc176 auto tensor_abs = input_abs->cast<abstract::AbstractTensorPtr>(); in SetInputs() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/optimizer/
Denviron_conversion.cc155 static const AbstractBasePtr tensor_abs = std::make_shared<abstract::AbstractTensor>(scalar_abs); in EnvironConversion() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/test/ut/src/runtime/
Dkernel_executor_tests.cc101 mindspore::MSTensor tensor_abs("Abs", mindspore::DataType::kNumberTypeFloat32, {1, 3, 2, 3}, in TEST_F() local
120 mindspore::MSTensor tensor_abs("Abs", mindspore::DataType::kNumberTypeFloat32, {1, 3, 2, 2}, in TEST_F() local
141 mindspore::MSTensor tensor_abs("Abs", mindspore::DataType::kNumberTypeFloat32, {1, 3, 2, 2}, in TEST_F() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/c_api/src/
Dutils.cc175 auto tensor_abs = BuildAbstract({shape}, {type}); in BuildAbstract() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/operator/composite/
Dtensor_index.cc423 auto tensor_abs = dyn_cast<abstract::AbstractTensor>(index_abs); in NormalizeTupleIndex() local
525 auto tensor_abs = index_abs->BuildValue()->cast<mindspore::tensor::TensorPtr>(); in ConstGetStrideInfoFromTuple() local
902 auto tensor_abs = dyn_cast<abstract::AbstractTensor>(index_abs); in FormatIndex() local
1225 auto tensor_abs = dyn_cast<abstract::AbstractTensor>(index_abs); in GenerateFuncGraph() local
/third_party/mindspore/mindspore-src/source/mindspore/core/abstract/ops/
Dprim_others.cc94 auto tensor_abs = abs_ref->ref(); in InferImplDepend() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/actor/
Dany_type_kernel_actor.cc328 const auto &tensor_abs = abstract->cast<abstract::AbstractTensorPtr>(); in GetElementType() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/pynative/forward/
Dforward.cc55 auto tensor_abs = op_run_info->base_op_run_info.abstract; in ShallowCopyValue() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/utils/
Dconvert_utils_py.cc73 auto tensor_abs = abs->cast<abstract::AbstractTensorPtr>(); in SetAdaptedAttrToTensor() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/jit/ps/static_analysis/
Devaluator.cc100 auto tensor_abs = arg->cast_ptr<AbstractTensor>(); in GetArgsUniqueDtype() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/
Dcontrol_node_parser.cc271 …const auto &tensor_abs = sequence_abs->dynamic_len_element_abs()->cast<abstract::AbstractTensorPtr… in FetchTypeIdByNode() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pybind_api/ir/
Dtensor_index_py.cc1810 auto tensor_abs = abs->cast<abstract::AbstractTensorPtr>(); in GetStubAbsTypeId() local