- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 53 for getNumArguments (0.2 sec)
-
tensorflow/compiler/mlir/lite/utils/nms_utils.cc
// Relevant argument characteristics are checked by the TFL op definition. if (func_.getNumArguments() < 5) { return func_.emitWarning() << "Invalid number of arguments to " "non_max_suppression_padded_v2 (need at least 5): " << func_.getNumArguments(); } if (func_.getFunctionType().getNumResults() != 2) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tf_saved_model_asset_sinking_pass.cc
void RewriteFunction(const SymbolTable& symbol_table, func::FuncOp func) { if (func.getNumArguments() == 0) { return; } auto builder = OpBuilder::atBlockBegin(&func.front()); llvm::SmallDenseMap<llvm::StringRef, TF::ConstOp> const_ops; llvm::BitVector arg_indexes_to_remove(func.getNumArguments()); // Replace arguments with const ops. for (BlockArgument argument : func.getArguments()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/perception_ops_utils.cc
} LogicalResult ConvertMaxUnpoolingFunc::VerifySignature() { // Verify high-level function signature. if (func_.getNumArguments() != 2) { return func_.emitWarning() << "Invalid number of arguments to " << kMaxUnpooling << ": " << func_.getNumArguments(); } if (func_.getFunctionType().getNumResults() != 1) { return func_.emitWarning()
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 8.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/optimize_global_tensors.cc
global_tensor.erase(); } } void EraseUnusedBoundInputs(ModuleOp module) { for (auto func : module.getOps<func::FuncOp>()) { llvm::BitVector args_to_erase(func.getNumArguments()); for (int i = 0, e = func.getNumArguments(); i < e; i++) { if (func.getArgAttr(i, "tf_saved_model.bound_input") && func.getArgument(i).use_empty()) { args_to_erase.set(i); } }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/insert_main_function.cc
if (input_names.size() != main_func.getNumArguments() || output_names.size() != main_func.getNumResults()) { module_op.emitError() << "Number of inputs and outputs in the tf.entry_function attribute " "mismatched. [Input] Expected: " << input_names.size() << ", got: " << main_func.getNumArguments() << ". [Output] Expected: " << output_names.size()
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 16.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tf_saved_model_freeze_variables.cc
if (!input_shapes_attr) return success(); if (input_shapes_attr.size() != func_op.getNumArguments()) { return func_op->emitError( "Number of arguments and 'tf._input_shapes' " "attribute size do not match. ") << "Num args: " << func_op.getNumArguments() << ", tf._input_shapes size: " << input_shapes_attr.size(); } return success(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 09:56:53 UTC 2024 - 19.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/promote_resources_to_args.cc
if (!return_op) return function.emitError() << "expects function '" << function.getName() << "' to have a MLIR ReturnOp"; llvm::SmallVector<ResourceInfo, 4> resources(function.getNumArguments()); auto argument_types = llvm::to_vector<4>(function.getFunctionType().getInputs()); bool has_resources = false; auto add_resource_argument = [&](BlockArgument arg,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 17.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/lstm_utils_test.cc
fused_lstm_func_.dump(); // verify transpose EXPECT_EQ( fused_lstm_func_->getAttrOfType<StringAttr>(kTFImplements).getValue(), convert.GetCompositeOpName()); EXPECT_EQ(fused_lstm_func_.getNumArguments(), 5); EXPECT_EQ(fused_lstm_func_.getFunctionType().getNumResults(), 1); auto transpose_op = fused_lstm_func_.getBody().front().begin(); transpose_op++;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo_conversions/reduce.cc
// return %1, %5 LogicalResult MatchReduceToArgMinMaxType2(mhlo::ReduceOp reduce_op, bool is_argmax) { Block& body = reduce_op.getBody().front(); if (body.getNumArguments() != 4) return failure(); mhlo::ReturnOp return_op = dyn_cast<mhlo::ReturnOp>(body.back()); if (!return_op || return_op.getNumOperands() != 2) return failure();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 05 20:53:17 UTC 2024 - 8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/lower_globals_to_ml_program.cc
// right away, since they might weave through blocks, but we can replace // them with a dummy which DCE can pick up later. llvm::BitVector argsToErase(func.getNumArguments()); for (int i = 0; i < func.getNumArguments(); i++) { if (auto global = tf_saved_model::LookupBoundInputOfType< tf_saved_model::GlobalTensorOp>(func, i, syms)) { OpBuilder builder(func.getBody());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.6K bytes - Viewed (0)