Searched defs:tensor_proto (Results 1 – 14 of 14) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/transform/express_ir/ |
| D | mindir_exporter.cc | 304 mind_ir::TensorProto *tensor_proto = value_proto->add_tensor(); in SetValueInfoProto() local
329 mind_ir::TensorProto *tensor_proto = attr_proto->add_tensors(); in SetTensorToAttributeProto() local
343 mind_ir::TensorProto *const tensor_proto) { in SetTensorProto()
355 …ilder::SetParamToTensorProto(const ParameterPtr ¶m, mind_ir::TensorProto *const tensor_proto) { in SetParamToTensorProto()
434 mind_ir::TensorProto *tensor_proto = attr_proto->add_tensors(); in SetShapeToNodeProto() local
443 mind_ir::TensorProto *tensor_proto = attr_proto->add_tensors(); in SetShapeToNodeProto() local
601 mind_ir::TensorProto *tensor_proto = attr_proto->add_tensors(); in SetTypeToAttributeProto() local
728 mind_ir::TensorProto *tensor_proto = attr_proto->add_tensors(); in SetScalarToAttributeProto_irs() local
733 mind_ir::TensorProto *tensor_proto = attr_proto->add_tensors(); in SetScalarToAttributeProto_irs() local
|
| D | onnx_exporter.cc | 585 …nnxExporter::SetTensorProtoInfo(const ParameterPtr ¶m, onnx::TensorProto *const tensor_proto) { in SetTensorProtoInfo()
1125 … onnx::TensorProto *const tensor_proto, const std::string tensor_name, in SetConstantNodeProtoInfoForGeLU()
1429 onnx::TensorProto *tensor_proto = attr_proto->mutable_t(); in ExportPrimDepthwiseConv2d() local
1538 onnx::TensorProto *tensor_proto = attr_proto->mutable_t(); in ExportPrimSquare() local
1920 … OnnxExporter::ConvertTupleToTensor(const ValuePtr &value, onnx::TensorProto *const tensor_proto) { in ConvertTupleToTensor()
1967 onnx::TensorProto *tensor_proto = attr_proto->mutable_t(); in SetNodeAttribute() local
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/tf/ |
| D | tf_model_parser.cc | 97 int GetShapeSize(const tensorflow::TensorProto &tensor_proto) { in GetShapeSize()
107 STATUS SetFloatTensorInfo(const tensorflow::TensorProto &tensor_proto, tensor::TensorPtr *tensor_in… in SetFloatTensorInfo()
142 STATUS SetInt32TensorInfo(const tensorflow::TensorProto &tensor_proto, tensor::TensorPtr *tensor_in… in SetInt32TensorInfo()
177 STATUS SetInt64TensorInfo(const tensorflow::TensorProto &tensor_proto, tensor::TensorPtr *tensor_in… in SetInt64TensorInfo()
220 STATUS SetBoolTensorInfo(const tensorflow::TensorProto &tensor_proto, tensor::TensorPtr *tensor_inf… in SetBoolTensorInfo()
248 STATUS SetStringTensorInfo(const tensorflow::TensorProto &tensor_proto, tensor::TensorPtr *tensor_i… in SetStringTensorInfo()
306 STATUS TFModelParser::ConvertConstVariant(const tensorflow::TensorProto &tensor_proto, tensor::Tens… in ConvertConstVariant()
371 STATUS TFModelParser::SetTensorInfoFromType(const tensorflow::TensorProto &tensor_proto, in SetTensorInfoFromType()
398 const tensorflow::TensorProto &tensor_proto = attr_value.tensor(); in ConvertConstTensor() local
|
| D | tf_resize_parser.cc | 65 auto tensor_proto = attr_value.tensor(); in Parse() local
|
| D | tf_concat_parser.cc | 41 auto tensor_proto = attr_value.tensor(); in Parse() local
|
| D | tf_tile_parser.cc | 33 const auto &tensor_proto = attr_value.tensor(); in Parse() local
|
| D | tf_gather_parser.cc | 50 auto tensor_proto = attr_value.tensor(); in Parse() local
|
| D | tf_pad_parser.cc | 54 auto &tensor_proto = attr_value.tensor(); in Parse() local
|
| D | tf_reverse_parser.cc | 41 auto tensor_proto = attr_value.tensor(); in Parse() local
|
| D | tf_slice_parser.cc | 39 auto tensor_proto = attr_value.tensor(); in Parse() local
|
| /third_party/mindspore/mindspore/profiler/parser/ |
| D | container.py | 200 def __init__(self, tensor_proto): argument
|
| /third_party/mindspore/mindspore/core/load_mindir/ |
| D | anf_model_parser.cc | 228 …TensorPtr MSANFModelParser::BuildTensorInfoForFuncGraph(const mind_ir::TensorProto &tensor_proto) { in BuildTensorInfoForFuncGraph()
314 const mind_ir::TensorProto &tensor_proto = value_proto.tensor(0); in BuildInputForFuncGraph() local
|
| /third_party/mindspore/mindspore/ccsrc/debug/ |
| D | dump_proto.cc | 220 irpb::TensorProto *tensor_proto = value_proto->mutable_tensor_val(); in SetValueToProto() local
|
| /third_party/mindspore/mindspore/ccsrc/debug/debugger/ |
| D | proto_exporter.cc | 156 debugger::TensorProto *tensor_proto = value_proto->mutable_tensor_val(); in SetValueToProto() local
|