| /third_party/mindspore/mindspore/lite/tools/converter/parser/tf/ |
| D | tf_conv_base_parser.cc | 37 STATUS TFConvBaseParser::ParseKernels(const tensorflow::NodeDef &node_def, const mindspore::Format … in ParseKernels() argument
41 if (!TensorFlowUtils::FindAttrValue(node_def, "value", &attr_value)) { in ParseKernels()
58 STATUS TFConvBaseParser::ParseStrides(const tensorflow::NodeDef &node_def, const mindspore::Format … in ParseStrides() argument
63 if (!TensorFlowUtils::FindAttrValue(node_def, "strides", &attr_value)) { in ParseStrides()
80 STATUS TFConvBaseParser::ParseExplicitPaddings(const tensorflow::NodeDef &node_def, const mindspore… in ParseExplicitPaddings() argument
84 if (!TensorFlowUtils::FindAttrValue(node_def, "explicit_paddings", &attr_value)) { in ParseExplicitPaddings()
107 STATUS TFConvBaseParser::ParseDilations(const tensorflow::NodeDef &node_def, const mindspore::Forma… in ParseDilations() argument
112 if (!TensorFlowUtils::FindAttrValue(node_def, "dilations", &attr_value)) { in ParseDilations()
129 mindspore::PadMode TFConvBaseParser::ParsePadMode(const tensorflow::NodeDef &node_def) { in ParsePadMode() argument
131 if (!TensorFlowUtils::FindAttrValue(node_def, "padding", &attr_value)) { in ParsePadMode()
|
| D | tf_conv_base_parser.h | 31 static STATUS ParseStrides(const tensorflow::NodeDef &node_def, const mindspore::Format &format,
33 static STATUS ParseDilations(const tensorflow::NodeDef &node_def, const mindspore::Format &format,
35 static STATUS ParseKernels(const tensorflow::NodeDef &node_def, const mindspore::Format &format,
37 …static STATUS ParseExplicitPaddings(const tensorflow::NodeDef &node_def, const mindspore::Format &…
39 static mindspore::PadMode ParsePadMode(const tensorflow::NodeDef &node_def);
|
| D | tf_model_parser.cc | 393 STATUS TFModelParser::ConvertConstTensor(const tensorflow::NodeDef &node_def, const tensorflow::Att… in ConvertConstTensor() argument
530 auto &node_def = tf_root_graph_->node(i); in Parse() local
531 tf_root_graph_nodes_[node_def.name()] = &node_def; in Parse()
532 tf_root_graph_nodes_vec_.emplace_back(&node_def); in Parse()
544 auto &node_def = tf_root_graph_->node(i); in Parse() local
545 status = ConvertOps(node_def, tf_root_graph_nodes_, func_graph, &anf_root_node_map_); in Parse()
626 auto &node_def = tf_sub_fuction.node_def(j); in ConvertSubgraphInputs() local
627 (*tf_sub_node_map)[node_def.name()] = &node_def; in ConvertSubgraphInputs()
628 subgraph_tf_node_vec.emplace_back(&node_def); in ConvertSubgraphInputs()
750 auto &node_def = tf_sub_fuction.node_def(j); in ConvertSubgraph() local
[all …]
|
| D | tf_util.cc | 50 bool TensorFlowUtils::FindAttrValue(const tensorflow::NodeDef &node_def, const std::string &attr_na… in FindAttrValue() argument
52 const google::protobuf::Map<std::string, tensorflow::AttrValue> &attr = node_def.attr(); in FindAttrValue()
61 TypeId TensorFlowUtils::ParseAttrDataType(const tensorflow::NodeDef &node_def, const std::string &a… in ParseAttrDataType() argument
63 if (!FindAttrValue(node_def, attr_name, &attr_value)) { in ParseAttrDataType()
143 mindspore::Format TensorFlowUtils::ParseNodeFormat(const tensorflow::NodeDef &node_def) { in ParseNodeFormat() argument
145 if (!FindAttrValue(node_def, "data_format", &attr_value)) { in ParseNodeFormat()
|
| D | tf_util.h | 33 static bool FindAttrValue(const tensorflow::NodeDef &node_def, const std::string &attr_name,
35 …static TypeId ParseAttrDataType(const tensorflow::NodeDef &node_def, const std::string &attr_name);
39 static mindspore::Format ParseNodeFormat(const tensorflow::NodeDef &node_def);
|
| D | tf_model_parser.h | 49 …static STATUS ConvertConstTensor(const tensorflow::NodeDef &node_def, const tensorflow::AttrValue …
59 …static STATUS ConvertInputNodes(const tensorflow::NodeDef &node_def, const std::vector<std::string…
66 STATUS ConvertOps(const tensorflow::NodeDef &node_def,
70 …ontrolFlowOp(const CNodePtr &anf_node, const string &op_type, const tensorflow::NodeDef &node_def);
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ge_runtime/task/ |
| D | aicpu_task.cc | 61 static_cast<uint32_t>(task_info_->node_def().size()) + sizeof(uint32_t); in Distribute()
151 auto size = task_info_->node_def().size(); in SetNodeDef()
161 … task_info_->node_def().size(), reinterpret_cast<const void *>(task_info_->node_def().data()), in SetNodeDef()
162 task_info_->node_def().size(), RT_MEMCPY_HOST_TO_DEVICE); in SetNodeDef()
|
| /third_party/mindspore/third_party/proto/tensorflow/ |
| D | function.proto | 6 import "node_def.proto";
58 // `node_def`) or as a return value of the function (in `ret`).
70 // * "node:out" where "node" is the name of a node in `node_def` and
91 // By convention, "op" in node_def is resolved by consulting with a
94 repeated NodeDef node_def = 3; field
97 // outputs from `node_def` that should be returned by the function.
101 // `node_def` which should be control outputs of this function.
|
| D | graph.proto | 6 import "node_def.proto";
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ge_runtime/ |
| D | task_info.h | 120 … const std::string &kernel_name, const std::string &node_def, const std::string &ext_info, in AicpuTaskInfo() argument
126 node_def_(node_def), in AicpuTaskInfo()
134 const std::string &node_def() const { return node_def_; } in node_def() function
|