- Sort Score
- Result 10 results
- Languages All
Results 21 - 30 of 58 for NamedAttribute (0.21 sec)
-
tensorflow/compiler/mlir/quantization/common/lift_as_function_call.h
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 10 04:07:09 UTC 2024 - 5.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/python/jax_to_tfl_flatbuffer.cc
input_names.push_back(input.name()); } const auto& inputs = absl::StrJoin(input_names, ","); mlir::OpBuilder builder(*module); llvm::SmallVector<mlir::NamedAttribute> attrs; attrs.push_back( builder.getNamedAttr("inputs", builder.getStringAttr(inputs))); // Jax wrapped the output nodes in a tuple, so it's pretty hard to us
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Mar 11 19:29:56 UTC 2024 - 8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/xla_call_module_serialization.cc
if (!called_func) { op->emitOpError() << "is missing attribute '" << kCalledFuncAttrName << "'"; return WalkResult::interrupt(); } llvm::SmallVector<NamedAttribute> new_config; // Copy the attributes in the current config except `called_func`. for (auto attr : backend_config) { if (attr.getName() != kCalledFuncAttrName) { new_config.push_back(attr); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 9.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/executor_tpuv1_outline_tpu_island.cc
// Outline all the transitively called functions by moving them in the // outlined module. for (func::FuncOp func : outlined_module.getOps<func::FuncOp>()) { func.walk([&](Operation *op) { for (NamedAttribute attr : op->getAttrs()) { if (auto symbol_ref = mlir::dyn_cast<FlatSymbolRefAttr>(attr.getValue())) { MoveFuncOp(symbol_ref, symbol_table, outlined_symbol_table); continue;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/export_utils.cc
return node_def; } Status ConvertAttributes( const llvm::ArrayRef<mlir::NamedAttribute> attrs, const absl::flat_hash_set<absl::string_view>& attrs_to_ignore, bool remove_ref_type, AttrValueMap* values) { AttrValueMap func_call_attrs; for (const mlir::NamedAttribute& named_attr : attrs) { auto name_strref = named_attr.getName().str(); auto attr = named_attr.getValue();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 19.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/export_graphdef.cc
if (auto device_attr = func.getArgAttrOfType<mlir::StringAttr>(index, kDeviceAttr)) *node_def->mutable_device() = device_attr.getValue().str(); llvm::ArrayRef<mlir::NamedAttribute> func_arg_i_attrs = mlir::function_interface_impl::getArgAttrs(func, index); absl::flat_hash_set<absl::string_view> attrs_to_ignore = {kDeviceAttr,
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
if (auto device_attr = func.getArgAttrOfType<mlir::StringAttr>(index, kDeviceAttr)) *node_def->mutable_device() = device_attr.getValue().str(); llvm::ArrayRef<mlir::NamedAttribute> func_arg_i_attrs = mlir::function_interface_impl::getArgAttrs(func, index); absl::flat_hash_set<absl::string_view> attrs_to_ignore = {kDeviceAttr,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 23:04:51 UTC 2024 - 35.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/device_util_test.cc
mlir::Builder builder(*module_ref); const std::string gpu0 = "/job:worker/replica:0/task:0/device:GPU:0"; const std::string gpu1 = "/job:worker/replica:0/task:0/device:GPU:1"; llvm::SmallVector<mlir::NamedAttribute, 2> metadata; metadata.push_back(builder.getNamedAttr( gpu0, mlir::TF::GpuDeviceMetadata::get(module_ref->getContext(), 1, 2))); (*module_ref)->setAttr("tf.devices", builder.getDictionaryAttr(metadata));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 9.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/integration/tfr_decompose_ctx.cc
TF_RETURN_IF_ERROR(ConvertDataType(ty, builder, &elt_ty)); mlir::TensorType mlir_ty = mlir::UnrankedTensorType::get(elt_ty); output_tys.push_back(mlir_ty); } llvm::SmallVector<mlir::NamedAttribute, 4> attrs; for (const auto& attr : node_def.attr()) { TF_ASSIGN_OR_RETURN(auto mlir_attr, ConvertAttributeValue(attr.second, &builder));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 29 02:34:43 UTC 2024 - 9.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/xla_call_module_deserialization.cc
<< " but enclosing XlaCallModule has a function list of size " << function_list.size(); return WalkResult::interrupt(); } llvm::SmallVector<NamedAttribute> new_config; // Copy the attributes in the current config except `called_index`. for (auto attr : backend_config) { if (attr.getName() != kCalledIndexAttrName) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 23 09:05:47 UTC 2024 - 11.1K bytes - Viewed (0)