Home
last modified time | relevance | path

Searched defs:tensor_proto (Results 1 – 16 of 16) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/mindir_loader/mindir_model/
Dmindir_model.h30 TensorProtoWrap(std::string name, const mind_ir::TensorProto &tensor_proto) in TensorProtoWrap()
34 const mind_ir::TensorProto &tensor_proto() { return tensor_proto_; } in tensor_proto() function
Dmindir_model_loader.cc152 const mind_ir::TensorProto &tensor_proto = graph_proto.input(i).tensor(0); in ConvertTensors() local
166 const mind_ir::TensorProto &tensor_proto = graph_proto.output(i).tensor(0); in ConvertTensors() local
180 const mind_ir::TensorProto &tensor_proto = graph_proto.parameter(i); in ConvertTensors() local
201 const mind_ir::TensorProto &tensor_proto = attribute_proto.tensors(0); in ConvertNodes() local
Dmindir_model_util.cc67 auto tensor_proto = attr_proto.tensors(0); in MakeValueFromTensorOrTypeAttribute() local
81 …e::ValuePtr MindirModelUtil::MakeValueFromTensorAttribute(const mind_ir::TensorProto &tensor_proto, in MakeValueFromTensorAttribute()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/common/debug/
Dmindir_exporter.cc611 mind_ir::TensorProto *tensor_proto = value_proto->add_tensor(); in SetValueInfoProto() local
632 mind_ir::TensorProto *tensor_proto = attr_proto->add_tensors(); in SetTensorToAttributeProto() local
678 …uilder::SetTensorProto(const AbstractBasePtr &abstract, mind_ir::TensorProto *const tensor_proto) { in SetTensorProto()
719 …ilder::SetParamToTensorProto(const ParameterPtr &param, mind_ir::TensorProto *const tensor_proto) { in SetParamToTensorProto()
1091 mind_ir::TensorProto *tensor_proto = attr_proto->add_tensors(); in SetAbstractToNodeProto() local
1095 mind_ir::TensorProto *tensor_proto = attr_proto->add_tensors(); in SetAbstractToNodeProto() local
1298 …Builder::SetTensorTypeToAttributeProto(const ValuePtr &value, mind_ir::TensorProto *tensor_proto) { in SetTensorTypeToAttributeProto()
1327 mind_ir::TensorProto *tensor_proto = attr_proto->add_tensors(); in SetTypeToAttributeProto() local
1446 auto tensor_proto = attr_proto->add_tensors(); in SetValueToAttributeProto() local
1528 mind_ir::TensorProto *tensor_proto = attr_proto->add_tensors(); in SetTypeToAttributeProto_irs() local
[all …]
Ddump_proto.cc234 irpb::TensorProto *tensor_proto = value_proto->mutable_tensor_val(); in SetValueToProto() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/parser/tf/
Dtf_concat_parser.cc43 auto tensor_proto = attr_value.tensor(); in Parse() local
Dtf_model_parser.cc109 int GetShapeSize(const tensorflow::TensorProto &tensor_proto) { in GetShapeSize()
119 STATUS SetFloatTensorInfo(const tensorflow::TensorProto &tensor_proto, tensor::TensorPtr *tensor_in… in SetFloatTensorInfo()
157 STATUS SetInt32TensorInfo(const tensorflow::TensorProto &tensor_proto, tensor::TensorPtr *tensor_in… in SetInt32TensorInfo()
204 STATUS SetBoolTensorInfo(const tensorflow::TensorProto &tensor_proto, tensor::TensorPtr *tensor_inf… in SetBoolTensorInfo()
233 STATUS SetStringTensorInfo(const tensorflow::TensorProto &tensor_proto, tensor::TensorPtr *tensor_i… in SetStringTensorInfo()
300 STATUS TFModelParser::SetInt64TensorToInt64Tensor(const tensorflow::TensorProto &tensor_proto, in SetInt64TensorToInt64Tensor()
336 STATUS TFModelParser::SetInt64TensorInfo(const tensorflow::TensorProto &tensor_proto, tensor::Tenso… in SetInt64TensorInfo()
346 STATUS TFModelParser::ConvertConstVariant(const tensorflow::TensorProto &tensor_proto, tensor::Tens… in ConvertConstVariant()
418 STATUS TFModelParser::SetTensorInfoFromType(const tensorflow::TensorProto &tensor_proto, tensor::Te… in SetTensorInfoFromType()
445 const tensorflow::TensorProto &tensor_proto = attr_value.tensor(); in ConvertConstTensor() local
Dtf_tile_parser.cc33 const auto &tensor_proto = attr_value.tensor(); in Parse() local
Dtf_gather_parser.cc52 auto tensor_proto = attr_value.tensor(); in Parse() local
Dtf_reverse_parser.cc41 auto tensor_proto = attr_value.tensor(); in Parse() local
Dtf_pad_parser.cc58 auto &tensor_proto = attr_value.tensor(); in Parse() local
Dtf_slice_parser.cc39 auto tensor_proto = attr_value.tensor(); in Parse() local
/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/profiler/parser/
Dcontainer.py200 def __init__(self, tensor_proto): argument
/third_party/mindspore/mindspore-src/source/mindspore/core/load_mindir/
Dload_model.cc426 mind_ir::TensorProto tensor_proto = attr_proto.tensors(0); in GetValueFromAttributeProto() local
693 …TensorPtr MSANFModelParser::GetAbsTensorFromTensorProto(const mind_ir::TensorProto &tensor_proto) { in GetAbsTensorFromTensorProto()
931 bool MSANFModelParser::GetTensorDataFromExternal(const mind_ir::TensorProto &tensor_proto, in GetTensorDataFromExternal()
1013 const mind_ir::TensorProto &tensor_proto = value_proto.tensor(0); in BuildInputForFuncGraph() local
1472 const mind_ir::TensorProto &tensor_proto = value.tensors(0); in ObtainValueInDictionaryForm() local
1532 mind_ir::TensorProto tensor_proto = elem_attr_proto.tensors(0); in ObtainValueInSequenceForm() local
1595 const auto &tensor_proto = attr_proto.tensors(0); in BuildValueFromAttributeProto() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/debug/debugger/
Dproto_exporter.cc158 debugger::TensorProto *tensor_proto = value_proto->mutable_tensor_val(); in SetValueToProto() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/transform/express_ir/
Donnx_exporter.cc2453 onnx::TensorProto *tensor_proto = attr_proto->mutable_t(); in ExportPrimDepthwiseConv2d() local
2558 onnx::TensorProto *tensor_proto = attr_proto->mutable_t(); in ExportPrimSquare() local
4733 void OnnxExporter::SetTensorData(const ValuePtr &value, onnx::TensorProto *tensor_proto) { in SetTensorData()