- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for hasOneBlock (0.28 sec)
-
tensorflow/compiler/mlir/tensorflow/utils/call_graph_util.cc
} callees.push_back(callee); } return success(); } bool HasSingleBlock(func::FuncOp func) { return func->getNumRegions() == 1 && func.getBody().hasOneBlock(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 2.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/region_control_flow_to_functional.cc
// The pattern we're matching is // ^block(arg0, arg1, ..., argN): // r0, r1, ..., rN = func.call @foo(arg0, arg1, ..., argN) // "tf.yield"(r0, r1, ..., rN) if (!region.hasOneBlock()) return std::nullopt; Block& block = region.front(); if (std::distance(block.begin(), block.end()) != 2) return std::nullopt; TF::YieldOp yield = llvm::dyn_cast_or_null<TF::YieldOp>(block.getTerminator());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 28.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc
auto function = dyn_cast_or_null<mlir::func::FuncOp>( SymbolTable::lookupSymbolIn(module, op.getReducer())); if (!function) return op.emitOpError() << "No reducer"; if (!function.getBody().hasOneBlock()) return op.emitOpError() << "reducer has more than one block"; return success(); } //===----------------------------------------------------------------------===// // XlaVariadicSortOp
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 22:07:10 UTC 2024 - 170.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo.cc
start_indices_tensor); return success(); }; }; template <typename ReturnOpType> bool MatchTopKComparator(Region& comparator) { if (!comparator.hasOneBlock()) return false; Block& comparator_blk = comparator.front(); using OpListType = llvm::iplist<Operation>; OpListType& operations = comparator_blk.getOperations(); if (operations.size() != 2) return false;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 154.9K bytes - Viewed (0)