- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 39 for BlockArgument (0.16 sec)
-
tensorflow/compiler/mlir/quantization/common/quantization_lib/quantization_driver.h
// result. void Finalize(); SmallVector<BlockArgument, 4> GetArgs() { return args_; } llvm::DenseMap<std::pair<mlir::Operation*, int>, int> GetResultStates() { return result_states_; } DenseMap<OpWithResultIndex, QuantStateIndex> result_states_; // Returns the state of the block argument. QuantState& GetArgQuantState(BlockArgument arg) { return states_[arg_states_[arg]]; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 20 11:42:17 UTC 2024 - 16.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/merge_save_function_ops_to_main.cc
// "inputs" key. BlockArgument CreateFilePrefixArg(func::FuncOp main_func_op) { Builder builder(main_func_op); // Add a new argument of type `tensor<!tf_type.string>` and update the // function type. auto file_prefix_arg_type = RankedTensorType::get(/*shape=*/{}, builder.getType<TF::StringType>()); BlockArgument new_file_prefix_arg =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_device_ops.td
unsigned GetNumPackedBlockArguments(); llvm::ArrayRef<BlockArgument> GetPackedBlockArguments(); llvm::ArrayRef<BlockArgument> GetReplicatedBlockArguments(); bool IsReplicatedBlockArgument(BlockArgument block_arg); bool IsPackedBlockArgument(BlockArgument block_arg); unsigned GetReplicaOperandIndexForBlockArgument(BlockArgument block_arg, unsigned replica);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jan 23 23:53:20 UTC 2024 - 14.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/push_transpose_through_ewise.cc
if (tpose_arg1_type.getRank() != tpose_arg2_type.getRank()) { return failure(); } if (llvm::isa<BlockArgument>(tpose_arg1.getPerm()) || llvm::isa<BlockArgument>(tpose_arg2.getPerm())) { return failure(); } auto perm1 = llvm::dyn_cast_or_null<arith::ConstantOp>( tpose_arg1.getPerm().getDefiningOp());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 12.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_space_to_depth_pass.cc
// Update input type. auto transform_result_type = RankedTensorType::get(new_shape, getElementTypeOrSelf(cast_input)); cast_input.setType(transform_result_type); auto block_arg = mlir::dyn_cast<mlir::BlockArgument>(cast_input); auto cast_op_input = dyn_cast_or_null<TF::CastOp>(cast_input.getDefiningOp()); while (block_arg || cast_op_input) { if (block_arg) { // Change on device function type/shape.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 29.3K 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/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/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/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)