Home
last modified time | relevance | path

Searched refs:op_list_ (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/ops/compat/
Dop_compatibility_lib.cc110 OpRegistry::Global()->Export(false, &op_list_); in OpCompatibilityLib()
120 RemoveDescriptionsFromOpList(&op_list_); in ValidateCompatible()
127 for (int cur = 0; iter != stable_ops_->end() && cur < op_list_.op_size(); in ValidateCompatible()
129 const string& op_name = op_list_.op(cur).name(); in ValidateCompatible()
157 while (cur < op_list_.op_size() && hist < in_op_history.size()) { in ValidateCompatible()
158 const OpDef& cur_op = op_list_.op(cur); in ValidateCompatible()
164 for (++cur; cur < op_list_.op_size(); ++cur) { in ValidateCompatible()
165 if (op_list_.op(cur).name() != cur_op_name) break; in ValidateCompatible()
234 for (; cur < op_list_.op_size(); ++cur) { in ValidateCompatible()
235 const string& op_name = op_list_.op(cur).name(); in ValidateCompatible()
[all …]
Dop_compatibility_lib.h55 string OpsString() const { return op_list_.DebugString(); } in OpsString()
59 int num_all_ops() const { return op_list_.op_size(); } in num_all_ops()
77 OpList op_list_; variable
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dmhlo_fusion.cc240 : op_list_(op_list), in FusionPlanner()
259 for (Operation* op : op_list_) { in Run()
262 if (!IsFusible(op_list_[node_id]) || in Run()
277 const SmallVectorImpl<Operation*>& op_list() const { return op_list_; } in op_list()
324 int num_nodes = op_list_.size(); in BuildNodeMap()
326 Operation* op = op_list_[node_id]; in BuildNodeMap()
413 if (!IsFusible(op_list_[node_to]) || !IsFusible(op_list_[node_from])) { in TryToContractEdge()
418 dyn_cast<InferFusibilityOpInterface>(op_list_[node_from]); in TryToContractEdge()
425 dyn_cast<InferFusibilityOpInterface>(op_list_[node_to]); in TryToContractEdge()
465 const SmallVectorImpl<Operation*>& op_list_; member in mlir::mhlo::__anon3c5f23730111::FusionPlanner