- Sort Score
- Result 10 results
- Languages All
Results 51 - 60 of 67 for NamedAttribute (0.2 sec)
-
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/translate/import_model.cc
DeferredConversionMetaData( const std::string& function_name, const std::vector<mlir::NamedAttribute>& attributes) : function_name(function_name), attributes(attributes) {} std::string function_name; std::vector<mlir::NamedAttribute> attributes; }; // Adds all the ordered_nodes to the shape refiner shape_refiner_. Then all
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 183.2K 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) -
tensorflow/compiler/mlir/lite/utils/lstm_utils.cc
return fused_func_op_.emitError() << "Invalid function attribute, expected " << kTFImplements << " attribute " "not found"; } // TODO(ashwinm, b/144775479): Make these NamedAttribute on TF import // once tf.function can support this. llvm::SmallVector<llvm::StringRef, 4> attr_tokens; attr.getValue().split(attr_tokens, ","); if (attr_tokens.empty()) { return fused_func_op_.emitError()
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 36.2K bytes - Viewed (0)