- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 66 for getNumArguments (0.27 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/freeze_saved_model_assets.cc
return; } SymbolTable symbol_table(module); for (auto func : module.getOps<func::FuncOp>()) { llvm::BitVector args_to_erase(func.getNumArguments()); OpBuilder builder(func.getBody()); for (int i = 0, e = func.getNumArguments(); i < e; ++i) { SmallVector<TF::InitializeTableFromTextFileV2Op, 4> init_table_from_text_file_ops_to_erase;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Jan 30 01:12:09 UTC 2023 - 4.3K 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/tensorflow/transforms/deduplicate_bound_input_bindings.cc
if (!mlir::tf_saved_model::IsExported(func)) return; llvm::SmallDenseMap<Attribute, unsigned, 8> unique_bound_inputs; llvm::BitVector arg_indices_to_erase(func.getNumArguments()); for (unsigned i = 0, e = func.getNumArguments(); i < e; i++) { auto attr = func.getArgAttrOfType<FlatSymbolRefAttr>( i, "tf_saved_model.bound_input"); if (!attr) continue;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 04 09:25:35 UTC 2022 - 2.4K 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/materialize_mlir_passthrough_op.cc
return; } if (main.getNumArguments() != op->getNumOperands()) { op->emitError() << "mismatch between MLIR Opaque Op number of operands (" << op->getNumOperands() << ") and main() entry point in the module (" << main.getNumArguments() << " args)\n"; return; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Nov 03 12:35:38 UTC 2022 - 4.2K 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)