Home
last modified time | relevance | path

Searched defs:const_tensor (Results 1 – 15 of 15) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/customize/op_proto/utils/
Dop_const.h51 Tensor const_tensor; in GetConstIntData() local
98 Tensor const_tensor; in GetConstInt() local
Dreduce_infer_util.cc158 Tensor const_tensor; in GetConstData() local
Dutil.cc710 bool GetConstValue(const Operator &op, const Tensor &const_tensor, const DataType &dtype, in GetConstValue()
734 bool GetConstValue(const Operator &op, const Tensor &const_tensor, const DataType &dtype, in GetConstValue()
749 bool GetScalerValue(const Operator &op, const Tensor &const_tensor, const DataType &dtype, std::int… in GetScalerValue()
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/parser/onnx/
Donnx_constant_parser.cc94 const auto &const_tensor = attr.t(); in Parse() local
Donnx_model_parser.cc621 for (const auto &const_tensor : onnx_graph.initializer()) { in CheckOnnxModel() local
783 for (const auto &const_tensor : onnx_graph.initializer()) { in SortOnnxNodeIndex() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/optimizer/irpass/
Dspecialize_transform.h80 auto &const_tensor = *(value->cast<tensor::TensorPtr>()); in NewReplaceValueNode() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/model/
Dgraph_builder.h102 tensor::TensorPtr const_tensor; in Tensor() local
136 auto const_tensor = std::make_shared<tensor::Tensor>(input); in Tensor() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/optimizer/
Dsoftmax_grad_fusion.cc56 auto const_tensor = value_node->cast<tensor::TensorPtr>(); in NeedFusion() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ir_fusion/
Dpadd_update_fusion.cc121 tensor::TensorPtr const_tensor = std::make_shared<tensor::Tensor>(kInt64->type_id(), const_shape); in Process() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/delegate/npu/
Dnpu_converter_utils.h107 ge::TensorPtr const_tensor = std::make_shared<hiai::Tensor>(const_tensor_desc); variable
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/base/
Dgroup_convolution_creator.cc160 auto const_tensor = CreateConstTensor(origin_inputs_.at(info.first), info.second, group_id); in NewConstTensor() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/customize/op_proto/
Dselection_ops_proto.cc328 static void GetUnsortedSegmentSumConstValue(const Tensor &const_tensor, const DataType &dtype, int6… in GetUnsortedSegmentSumConstValue()
440 static void GetSliceConstValue(const Tensor &const_tensor, const DataType &dtype, std::vector<int64… in GetSliceConstValue()
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/delegate/tensorrt/op/
Delementwise_tensorrt.cc338 auto &const_tensor = in_tensors_[const_tensor_index]; in AddConstTensor() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/
Druntime_shape_fusion_pass.cc301 auto const_tensor = src_tensors_->at(input_index); in GenerateFusedShapeFusionMatrix() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/session/
Danf_runtime_algorithm.cc2107 const auto &const_tensor = values[0]->cast<tensor::TensorPtr>(); in CheckValidTensorTuple() local