- Sort Score
- Result 10 results
- Languages All
Results 1 - 7 of 7 for LookupBoundInputOfType (0.27 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/cc/convert_asset_args.cc
#include "tensorflow/core/protobuf/meta_graph.pb.h" namespace mlir::quant { namespace { using ::mlir::tf_saved_model::AssetOp; using ::mlir::tf_saved_model::kTfSavedModelIndexPathAttr; using ::mlir::tf_saved_model::LookupBoundInputOfType; using ::tensorflow::AssetFileDef; // Given argument attributes `arg_attrs`, returns a new set of argument // attributes where the "tf_saved_model.bound_input" attribute has been replaced
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/lower_globals_to_ml_program.cc
llvm::DenseSet<Value> seen; BlockArgument arg; if (traceUpwardsToArgument(resource, seen, &arg).failed()) return nullptr; Operation *global = tf_saved_model::LookupBoundInputOfType<tf_saved_model::GlobalTensorOp>( func, arg.getArgNumber(), syms); auto name = opToName[global]; if (name.empty()) return nullptr; return SymbolRefAttr::get(func->getContext(), opToName[global]); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/analysis/resource_dataflow.cc
if (func::FuncOp func = dyn_cast<func::FuncOp>(barg.getOwner()->getParentOp())) { SymbolTable symbol_table(func->getParentOfType<ModuleOp>()); auto global_tensor = tf_saved_model::LookupBoundInputOfType< tf_saved_model::GlobalTensorOp>(func, barg.getArgNumber(), symbol_table); ResourceConstructingOps result(global_tensor); return result; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_saved_model.h
// refers to as a bound input, or null. Operation *LookupBoundInput(func::FuncOp func, int arg_index, const SymbolTable &symbol_table); template <typename T> T LookupBoundInputOfType(func::FuncOp func, int arg_index, const SymbolTable &symbol_table) { return llvm::dyn_cast_or_null<T>( LookupBoundInput(func, arg_index, symbol_table)); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jan 18 03:21:34 UTC 2024 - 5.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tf_saved_model_asset_sinking_pass.cc
llvm::BitVector arg_indexes_to_remove(func.getNumArguments()); // Replace arguments with const ops. for (BlockArgument argument : func.getArguments()) { auto asset = LookupBoundInputOfType<AssetOp>( func, argument.getArgNumber(), symbol_table); if (asset == nullptr) { continue; } // Create a const op for the asset if it doesn't already exist.
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/lift_variables.cc
SymbolTable symbol_table(module); for (auto func : module.getOps<func::FuncOp>()) { for (auto arg : func.getArguments()) { unsigned arg_number = arg.getArgNumber(); auto global_tensor = LookupBoundInputOfType<GlobalTensorOp>( func, arg_number, symbol_table); if (!global_tensor) continue; auto arg_type = mlir::cast<RankedTensorType>(arg.getType()); assert(arg_type.getRank() == 0);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 23 09:05:47 UTC 2024 - 7.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/import_model.cc
llvm::SmallVector<mlir::Type, 4> new_input_types; for (int i = 0, e = func.getNumArguments(); i < e; i++) { auto arg = func.getArgument(i); auto global_tensor = mlir::tf_saved_model::LookupBoundInputOfType< mlir::tf_saved_model::GlobalTensorOp>(func, i, symbol_table); if (global_tensor) { auto old_type = arg.getType(); auto new_type =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 183.2K bytes - Viewed (0)