- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 46 for device_str (0.3 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/extract_tpu_copy_with_dynamic_shape_op.cc
op->emitError() << "TPUCopyWithDynamicShapeOp is not in a launch"; } std::string device_str = launch_op.getDeviceAttr().getValue().str(); std::string cpu0_device; if (failed(tensorflow::GetNonReplicatedCPU0(op, &cpu0_device))) return failure(); if (device_str != tensorflow::GetDeviceAliasForHostOfLogicalCore(0) && device_str != cpu0_device) { op->emitError()
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_a_m.cc
auto device_attr = op->getAttrOfType<StringAttr>("device"); if (!device_attr || device_attr.getValue().empty()) return device_ordinal_str; // TODO(b/229028654) Remove string conversion once implicit conversion between // llvm::StringRef and absl::string_view works. absl::string_view device_str(device_attr.data(), device_attr.size());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 146.7K bytes - Viewed (0) -
tensorflow/c/experimental/stream_executor/stream_executor_test.cc
SP_DeviceMemoryBase* const device_dst, const SP_DeviceMemoryBase* const device_src, uint64_t size, TF_Status* const status) { TF_SetStatus(status, TF_OK, ""); std::memcpy(device_dst->opaque, device_src->opaque, size); }; StreamExecutor* executor = GetExecutor(0);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 20 19:54:04 UTC 2024 - 26.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/mlir_graph_optimization_pass_test.cc
::testing::EquivToProto(original_graph_def))); #endif } void InitGraphOptions() { session_options_.config = config_proto_; graph_optimization_pass_options_.device_set = &device_set_; graph_optimization_pass_options_.session_options = &session_options_; graph_optimization_pass_options_.graph = &graph_; graph_optimization_pass_options_.flib_def = flib_.get(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Feb 27 08:25:30 UTC 2024 - 16.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_device_propagation.cc
// If op already has a TPU device set, simply propagate its device. auto device_attr = op_to_update->getAttrOfType<StringAttr>(kDeviceAttr); const bool has_device = device_attr && !device_attr.getValue().empty(); if (has_device && tensorflow::IsTPUDevice(device_attr.getValue())) { PopulateDeviceForOpResults(*op_to_update, device_attr.getValue(), value_to_device); continue;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11K bytes - Viewed (0) -
tensorflow/compiler/jit/device_util_test.cc
string* result) { jit::DeviceInfoCache cache; jit::DeviceSet device_set; for (absl::string_view name : device_names) { TF_ASSIGN_OR_RETURN(jit::DeviceId device_id, cache.GetIdFor(name)); device_set.Insert(device_id); } TF_ASSIGN_OR_RETURN( jit::DeviceId result_id, PickDeviceForXla(cache, device_set, allow_mixing_unknown_and_cpu)); *result = string(cache.GetNameFor(result_id));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/cluster_formation.cc
} std::string GetDevice(Operation* op) { auto device_attr = op->getAttrOfType<StringAttr>("device"); return device_attr ? device_attr.getValue().str() : ""; } bool CanBeIgnoredInCluster(Operation* op) { auto device_attr = op->getAttrOfType<StringAttr>("device"); return !device_attr || device_attr.getValue().empty(); } void BuildClusters(Block& block, OpBuilder builder,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Oct 05 13:30:21 UTC 2023 - 6.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/device_util.cc
} } // namespace void AddDevicesToOp(mlir::Operation* op, const DeviceSet* device_set) { if (!device_set) return; mlir::MLIRContext* ctx = op->getContext(); mlir::Builder builder(ctx); // Collect devices with attached metadata. llvm::SmallVector<mlir::NamedAttribute, 8> devices; devices.reserve(device_set->devices().size());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 6.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/cluster_util_test.cc
} return mlir_module; } std::string GetDevice(Operation* op) { auto device_attr = op->getAttrOfType<StringAttr>("device"); return device_attr ? device_attr.getValue().str() : ""; } bool CanBeIgnoredInCluster(Operation* op) { auto device_attr = op->getAttrOfType<StringAttr>("device"); return !device_attr || device_attr.getValue().empty(); } llvm::StringMap<SmallVector<Cluster>> GetClusters(ModuleOp module) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Apr 26 09:37:10 UTC 2024 - 7.8K bytes - Viewed (0) -
tensorflow/c/experimental/saved_model/core/test_utils.cc
DeviceFactory::NewDevice("CPU", {}, "/job:localhost/replica:0/task:0")); } EagerContextPtr CreateTestingEagerContext(DeviceMgr* device_mgr) { return EagerContextPtr(new EagerContext( SessionOptions(), tensorflow::ContextDevicePlacementPolicy::DEVICE_PLACEMENT_SILENT, /* async= */ false, device_mgr, /* device_mgr_owned= */ false, /* rendezvous= */ nullptr, /* cluster_flr= */ nullptr)); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Sep 08 20:13:32 UTC 2023 - 5.8K bytes - Viewed (0)