- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 73 for BlockArgument (0.8 sec)
-
tensorflow/compiler/mlir/tensorflow/analysis/tf_dataflow.h
while_op, while_op.getBodyAttr()); for (auto &arg : while_op->getOpOperands()) { BlockArgument cond_arg = cond.getArgument(arg.getOperandNumber()); this->join(getLatticeElement(cond_arg), *getLatticeElement(arg.get())); BlockArgument body_arg = body.getArgument(arg.getOperandNumber()); this->join(getLatticeElement(body_arg), *getLatticeElement(arg.get())); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 06 23:53:00 UTC 2024 - 3.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/promote_resources_to_args.cc
// `tf.ReadVariableOp` and `tf.AssignVariableOp` only. mlir::LogicalResult ValidateResourceArgument(func::FuncOp function, BlockArgument resource_arg, TF::ResourceType resource_type) { if (resource_type.getSubtypes().size() != 1) return function.emitError() << "expects resource type of argument "
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/tensorflow/transforms/freeze_global_tensors.cc
} } // Collect all those freezable. This is an extra scan but allows for the // partial behavior from `allow_mutable_tensor`. DenseMap<BlockArgument, bool> freezeable; for (auto func : module.getOps<func::FuncOp>()) { for (BlockArgument val : func.getArguments()) { if (!mlir::isa<TF::ResourceType>(getElementTypeOrSelf(val.getType()))) continue;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_dynamic_layout_pass.cc
return false; llvm::SmallSetVector<Value, 8> aliases = resource_alias_analysis.GetResourceAliases(resource_iterator); auto is_generator = [](Value val) { if (mlir::isa<BlockArgument>(val)) return true; Operation* definition = val.getDefiningOp(); return definition->getNumOperands() == 0 && definition->getNumResults() == 1; };
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 12.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/op_or_arg_name_mapper.cc
result.getResultNumber()); return std::string(result.getOwner()->getName().getStringRef()); } // Use the ASM syntax for BlockArgument if (auto arg = mlir::dyn_cast<mlir::BlockArgument>(val)) { return "arg" + std::to_string(arg.getArgNumber()); } return ""; } std::string OpOrArgStripNameMapper::GetName(OpOrVal op_or_val) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tensor_device_copy_conversion.cc
// defining op. if (def_op != nullptr && (isa<TF::TPUExecuteOp, TF::TPUExecuteAndUpdateVariablesOp>(def_op))) { return true; } if (BlockArgument block_arg = mlir::dyn_cast<BlockArgument>(arg)) { // Skip the folding logic if the block argument is not from the function // arguments. This can happen when the argument is from a while loop.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 4.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_device.cc
// replicate op is being modified. llvm::ArrayRef<BlockArgument> ReplicateOp::GetReplicatedBlockArguments() { return GetBody().getArguments().drop_back(GetNumPackedBlockArguments()); } // Returns the packed block arguments. A copy should be made if the replicate op // is being modified. llvm::ArrayRef<BlockArgument> ReplicateOp::GetPackedBlockArguments() {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 33.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/hoist_broadcast_read.cc
#include "tensorflow/compiler/mlir/tensorflow/ir/tf_ops.h" #include "tensorflow/core/util/device_name_utils.h" namespace tensorflow { namespace tf2xla { namespace internal { namespace { using mlir::BlockArgument; using mlir::failure; using mlir::LogicalResult; using mlir::Operation; using mlir::OperationPass; using mlir::OpOperand; using mlir::StringAttr; using mlir::success; using mlir::Value;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/export_graphdef.cc
dst_index); } return absl::OkStatus(); } auto input_arg = mlir::cast<BlockArgument>(src); auto input_node_it = args_.find(input_arg); TF_RET_CHECK(input_node_it != args_.end()) << "Use of BlockArgument encounted before def!"; // For argument, there is only one result output, so the index is always 0.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 35.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v2/tf_executor_to_graph.cc
dst_index); } return absl::OkStatus(); } auto input_arg = mlir::cast<BlockArgument>(src); auto input_node_it = args_.find(input_arg); TF_RET_CHECK(input_node_it != args_.end()) << "Use of BlockArgument encounted before def!"; // For argument, there is only one result output, so the index is always 0.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 23:04:51 UTC 2024 - 35.2K bytes - Viewed (0)