- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for hasOneBlock (0.43 sec)
-
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)