- Sort Score
- Result 10 results
- Languages All
Results 21 - 30 of 31 for NamedAttribute (0.25 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/passes/insert_main_function.cc
module_op.getLoc(), kImportModelDefaultGraphFuncName, func_type); builder.createBlock(&main_func.getBody(), main_func.begin(), arg_types, arg_locs); SmallVector<NamedAttribute> func_attrs; func_attrs.push_back( {StringAttr::get(context, "inputs"), StringAttr::get(context, absl::StrJoin(input_names, ","))}); func_attrs.push_back( {StringAttr::get(context, "outputs"),
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/tf2xla/internal/passes/tpu_cluster_formation.cc
namespace tensorflow { namespace tf2xla { namespace internal { namespace { using mlir::Block; using mlir::DialectRegistry; using mlir::LogicalResult; using mlir::ModuleOp; using mlir::NamedAttribute; using mlir::NamedAttrList; using mlir::OpBuilder; using mlir::Operation; using mlir::OpResult; using mlir::Region; using mlir::StringAttr; using mlir::success; using mlir::Type; using mlir::Value;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 22:03:30 UTC 2024 - 39.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/ir/tfr_ops.td
let regions = (region VariadicRegion<AnyRegion>:$body); let skipDefaultBuilders = 1; let builders = [ OpBuilder<(ins "StringRef":$name, "FunctionType":$type, CArg<"ArrayRef<NamedAttribute>", "{}">:$attrs)> ]; let extraClassDeclaration = [{ LogicalResult verifyType() { return success(); } mlir::Region *getCallableRegion();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 22 10:54:29 UTC 2024 - 17.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/legalize_tf.cc
// original ophint converted node's 3rd output. SmallVector<Type, 4> result_types; result_types.push_back(op->getOpResult(2).getType()); // Populate attributes. SmallVector<NamedAttribute, 4> attributes; // Activation will always be tanh. attributes.push_back(rewriter.getNamedAttr("fused_activation_function", rewriter.getStringAttr("TANH")));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 20 20:06:54 UTC 2024 - 45.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/lift_as_function_call_test.cc
func::FuncOp main_fn = FindMainFuncOp(*module_op); ASSERT_THAT(main_fn, NotNull()); auto dot_general_op = FindOperationOfType<mlir::stablehlo::DotGeneralOp>(main_fn); const SmallVector<NamedAttribute>& attributes = { builder_.getNamedAttr( "precision_config", builder_.getArrayAttr(SmallVector<Attribute>( 1, mlir::stablehlo::PrecisionAttr::get(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 10 04:07:09 UTC 2024 - 26.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/tpu_merge_variables_with_execute.cc
auto merged_execute = builder->create<TF::TPUExecuteAndUpdateVariablesOp>( execute_launch.getLoc(), new_output_types, var_access_info.new_operand_values, llvm::ArrayRef<NamedAttribute>{ builder->getNamedAttr( "device_var_reads_indices", builder->getI64ArrayAttr(device_var_reads_indices)), builder->getNamedAttr(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 29 17:52:11 UTC 2024 - 27K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/resource_op_lifting_cleanup.cc
func::FuncOp cloned = CloneFunctionIfNeeded(func); cloned_branches.push_back(cloned); if (cloned == func) continue; // Patch up the op attribute to point to the new function. for (NamedAttribute attr : op->getAttrs()) { auto symref = mlir::dyn_cast<FlatSymbolRefAttr>(attr.getValue()); if (!symref) continue; if (symref.getValue() != func.getName()) continue; op->setAttr(attr.getName(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 19.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/tpu_variable_runtime_reformatting.cc
{}, TF::ResourceType::get(llvm::ArrayRef<TensorType>{key_type}, builder->getContext()))}, llvm::ArrayRef<Value>{}, llvm::ArrayRef<NamedAttribute>{ builder->getNamedAttr(kDeviceAttr, builder->getStringAttr(device)), builder->getNamedAttr("container", builder->getStringAttr("")), builder->getNamedAttr(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 21.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/executor_tpuv1_island_coarsening.cc
auto funcs_for_cluster = tpu_funcs.find(cluster); assert(funcs_for_cluster != tpu_funcs.end()); assert(!funcs_for_cluster->second.empty()); if (funcs_for_cluster->second.size() == 1) return false; for (NamedAttribute attr : op->getAttrs()) { auto symbol_ref = mlir::dyn_cast<FlatSymbolRefAttr>(attr.getValue()); if (!symbol_ref) continue; func::FuncOp callee =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 27.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/tpu_rewrite_pass.cc
tpu_devices, OpBuilder* builder) { if (!replicate) return; const int num_replicas = tpu_devices.size(); const int num_cores_per_replica = tpu_devices.front().size(); llvm::SmallVector<NamedAttribute, 8> device_attrs; for (int core = 0; core < num_cores_per_replica; ++core) { llvm::SmallVector<StringRef, 8> devices_by_core; devices_by_core.reserve(num_replicas);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 21:25:12 UTC 2024 - 29.7K bytes - Viewed (0)