- Sort Score
- Result 10 results
- Languages All
Results 41 - 50 of 73 for BlockArgument (0.28 sec)
-
tensorflow/compiler/mlir/tf2xla/internal/passes/tpu_sharding_identification_pass.cc
using OpShardingVector = llvm::SmallVector<OpShardingVariant, 8>; using OptionalOpShardingVector = llvm::SmallVector<std::optional<OpShardingVariant>, 8>; using llvm::StringRef; using mlir::Block; using mlir::BlockArgument; using mlir::BoolAttr; using mlir::Builder; using mlir::IntegerAttr; using mlir::LogicalResult; using mlir::ModuleOp; using mlir::Operation; using mlir::OpOperand; using mlir::OpResult;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 02:01:13 UTC 2024 - 28.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/resource_device_inference.cc
continue; } // Transfer the annotation to both region arguments for (Region* region : while_region.getRegions()) { BlockArgument arg = region->getArgument(output_index); LLVM_DEBUG(llvm::dbgs() << " Propagating device = '" << *device << "' to arg #" << output_index << " of region #"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 03 03:47:00 UTC 2023 - 13.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/cc/convert_asset_args.cc
if (!main_func_op) return failure(); SmallVector<StringRef> input_names = GetEntryFunctionInputs(main_func_op); SymbolTable symbol_table(module_op); SmallVector<AssetFileDef> asset_file_defs; for (BlockArgument argument : main_func_op.getArguments()) { const int arg_idx = argument.getArgNumber(); auto asset_op = LookupBoundInputOfType<AssetOp>(main_func_op, arg_idx, symbol_table); if (!asset_op) continue;
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/sparsecore/embedding_sequencing.cc
// args. This will basically be our set increment fn. for (Operation* op : non_tpu_ops) for (Value input : op->getOperands()) if (BlockArgument arg = llvm::dyn_cast<BlockArgument>(input)) // TODO(bfontain): Check that this is actually an argument to the loop // body. non_tpu_args.insert(arg.getArgNumber());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 39.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/cluster_tf_ops_pass.cc
std::string value_device = ""; // If the value is defined as an argument of the func_op, adds it to // the argument list of the function that uses this op. if (BlockArgument block_arg = mlir::dyn_cast<BlockArgument>(value)) { if (StringAttr attr = func_op.getArgAttrOfType<StringAttr>( block_arg.getArgNumber(), kTFDeviceAttr)) { value_device = attr.getValue().str(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 13.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tf_saved_model_asset_sinking_pass.cc
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()) { auto asset = LookupBoundInputOfType<AssetOp>( func, argument.getArgNumber(), symbol_table); if (asset == nullptr) { continue; }
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/quantization/tensorflow/passes/prepare_quantize.cc
return (arg.hasOneUse() && llvm::isa<quantfork::QuantizeCastOp>(*arg.user_begin())); }; bool need_to_set_input_nodes_quantization_params = false; for (const BlockArgument arg : func.getArguments()) { auto shaped = mlir::dyn_cast<ShapedType>(arg.getType()); if (shaped && mlir::isa<FloatType>(shaped.getElementType()) && !has_quantize_op(arg)) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 17.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/resource_op_lifting.cc
// is read in the function body. These read values are just the same argument // with type replaced. llvm::SmallVector<Value, 4> skipped_args; for (auto& it : hoister.GetResources()) { BlockArgument arg = mlir::dyn_cast<BlockArgument>(it.first); assert(arg && "Expect resources for FuncOp to be its arguments"); auto type_iter = resource_data_types.find(arg.getArgNumber()); if (type_iter == resource_data_types.end()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 55.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tensor_array_ops_decomposition.cc
llvm::SmallDenseMap<int64_t, llvm::StringSet<>> result_sets; auto insert = [&](Value v, const string& source, const Block& func_block) { auto arg = v.dyn_cast<BlockArgument>(); if (!arg || arg.getOwner() != &func_block) return; auto insert_res = result_sets[arg.getArgNumber()].insert(source); if (!insert_res.second) return; result[arg.getArgNumber()].push_back(source);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 02 20:41:19 UTC 2023 - 40.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/prepare_quantize.cc
return (arg.hasOneUse() && llvm::isa<quantfork::QuantizeCastOp>(*arg.user_begin())); }; bool need_to_set_input_nodes_quantization_params = false; for (const BlockArgument arg : func.getArguments()) { auto shaped = mlir::dyn_cast<ShapedType>(arg.getType()); if (shaped && mlir::isa<FloatType>(shaped.getElementType()) && !has_quantize_op(arg)) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 17.6K bytes - Viewed (0)