- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 70 for StringAttr (0.17 sec)
-
tensorflow/compiler/mlir/tensorflow/ir/tf_saved_model.cc
// Stores previously seen init types. llvm::SmallDenseSet<StringAttr> init_types{}; for (auto init_sym : session_init_op.getInitializers().getAsValueRange<FlatSymbolRefAttr>()) { auto init_func = symbol_table.lookup<func::FuncOp>(init_sym); if (!init_func) continue; auto init_type = init_func->getAttrOfType<StringAttr>(kTfSavedModelInitializerTypeAttr); if (!init_type) continue;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 23.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/lstm_utils_test.cc
attributes.push_back(kLayerNormalizedLstmCellSimple); } else { attributes.push_back(kLstmCellSimple); } if (cifg) { attributes.push_back(kCoupleInputForgetGates); } mlir::StringAttr attr_values = builder->getStringAttr(llvm::join(attributes, ",")); func->setAttr(kTFImplements, attr_values); return func; } class LstmUtilsTest : public ::testing::Test { protected:
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/xla_call_module_deserialization.cc
std::vector<std::string> disabled_checks; for (auto attr : op.getDisabledChecks().getAsRange<StringAttr>()) { disabled_checks.push_back(attr.getValue().str()); } std::vector<std::string> platforms; for (auto attr : op.getPlatforms().getAsRange<StringAttr>()) { platforms.push_back(attr.getValue().str()); } TF_ASSIGN_OR_RETURN( auto loader,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 23 09:05:47 UTC 2024 - 11.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/experimental/tac/transforms/raise_target_subgraphs.cc
StringAttr interface_name = builder.getStringAttr(absl::StrCat("func_", func_count)); added_func_op->setAttr(kInterfaceNameAttr, interface_name); added_call_op->setAttr(kInterfaceNameAttr, interface_name); StringAttr device = mlir::cast<StringAttr>( added_func_op->getRegion(0).getBlocks().front().front().getAttr(kDevice)); StringAttr inference_type = mlir::cast<StringAttr>(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops.cc
using DeviceNameUtils = tensorflow::DeviceNameUtils; // Duplicate of the logic in MultiDeviceFunctionBodyPlacer::BodyNodeDevice // LINT.IfChange auto device_id = StringAttr::get(call->getContext(), "device"); auto caller_device = call->getAttrOfType<StringAttr>(device_id); if (!caller_device) return; DeviceNameUtils::ParsedName caller_parsed_device; if (!DeviceNameUtils::ParseFullName(caller_device.getValue().str(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 14.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_dynamic_layout_pass.cc
if (!is_generator(alias)) return true; StringAttr device; if (auto arg = mlir::dyn_cast<BlockArgument>(alias)) { device = func.getArgAttrOfType<mlir::StringAttr>(arg.getArgNumber(), kFuncDeviceAttr); } else { device = alias.getDefiningOp()->getAttrOfType<StringAttr>(kDeviceAttr); } if (!device) return false;
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/utils/xla_sharding_util.cc
if (!mlir::isa<mlir::StringAttr>(shard_attr)) return mlir::failure(); auto shard_str = mlir::cast<mlir::StringAttr>(shard_attr).getValue().str(); return DecodeShardingAttribute(shard_str, sharding, report_error); } void EncodeSharding(mlir::Operation* op, llvm::StringRef shard_str) { if (!op->hasAttrOfType<mlir::StringAttr>(shard_str)) return; ::xla::OpSharding sharding;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 22 21:28:13 UTC 2024 - 34K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/export_graphdef.cc
if (dict_attr) { TF_RET_CHECK(mlir::isa<mlir::StringAttr>(dict_attr.get("inputs"))) << "inputs missing in entry function attribute"; TF_RET_CHECK(mlir::isa<mlir::StringAttr>(dict_attr.get("outputs"))) << "outputs missing in entry function attribute"; mlir::cast<mlir::StringAttr>(dict_attr.get("inputs")) .getValue()
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 (dict_attr) { TF_RET_CHECK(mlir::isa<mlir::StringAttr>(dict_attr.get("inputs"))) << "inputs missing in entry function attribute"; TF_RET_CHECK(mlir::isa<mlir::StringAttr>(dict_attr.get("outputs"))) << "outputs missing in entry function attribute"; mlir::cast<mlir::StringAttr>(dict_attr.get("inputs")) .getValue()
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/tf2xla/internal/passes/xla_broadcast.cc
block_arg.getLoc(), DenseIntElementsAttr::get( RankedTensorType::get({1, num_replicas}, builder.getIntegerType(32)), group_assignment_val)); StringAttr reduce_op = builder.getStringAttr("Add"); StringAttr mode = builder.getStringAttr("CrossReplica"); return builder.create<XlaAllReduceOp>(block_arg.getLoc(), block_arg.getType(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 13 18:52:07 UTC 2024 - 13.9K bytes - Viewed (0)