Searched refs:op_def (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore/lite/src/ops/ |
| D | ops_def.h | 56 std::string op_def = "\ntable "; \
57 op_def.append(#OP); \
58 op_def.append(" {\n");
73 #define OP_ATTR(key, type) op_def.append(" ").append(#key).append(": ").append(#type).append(";\…
74 #define OP_ATTR_ENUM(key, type) op_def.append(" ").append(#key).append(": ").append(#type).appen…
75 #define OP_ATTR_VEC2D(key, type) op_def.append(" ").append(#key).append(": ").append(#type).appe…
113 …op_def.append(" ").append(#key).append(": ").append(#type).append(" = ").append(#value).append(…
115 …op_def.append(" ").append(#key).append(": ").append(#type).append(" = ").append(#value).append(…
133 op_def.append("}\n"); \
134 return op_def; \
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/mem_reuse/ |
| D | mem_reuse_checker.cc | 342 void MemReuseChecker::SetTesnorFromAndToInfo(const KernelDef *op_def) { in SetTesnorFromAndToInfo() argument
343 auto split_name = GetSplitName(op_def->scope_full_name()); in SetTesnorFromAndToInfo()
344 for (auto &in : op_def->inputs_) { in SetTesnorFromAndToInfo()
351 for (auto &ou : op_def->outputs_) { in SetTesnorFromAndToInfo()
414 void MemReuseChecker::SetMembuInfos(const KernelDef *op_def, const std::vector<MembufPtr> &membuf_p… in SetMembuInfos() argument
422 auto split_name = GetSplitName(op_def->scope_full_name()); in SetMembuInfos()
424 SetTesnorFromAndToInfo(op_def); in SetMembuInfos()
427 void MemReuseChecker::SetAddNewMembuInfos(const KernelDef *op_def, const std::vector<MembufPtr> &me… in SetAddNewMembuInfos() argument
437 auto split_name = GetSplitName(op_def->scope_full_name()); in SetAddNewMembuInfos()
440 add_new_stream_ids_.push_back(op_def->stream_id()); in SetAddNewMembuInfos()
|
| D | mem_reuse_checker.h | 50 void SetMembuInfos(const KernelDef *op_def, const std::vector<MembufPtr> &membuf_ptr_list);
51 void SetTesnorFromAndToInfo(const KernelDef *op_def);
58 …void SetAddNewMembuInfos(const KernelDef *op_def, const std::vector<MembufPtr> &membuf_ptr_list, s…
|
| /third_party/toybox/toys/pending/ |
| D | expr.c | 135 static struct op_def { struct
154 void eval_op(struct op_def *o, struct value *ret, struct value *rhs) in eval_op() argument
231 struct op_def *o = OPS; in eval_expr()
|
| /third_party/mindspore/third_party/proto/tensorflow/ |
| D | function.proto | 7 import "op_def.proto";
|