- Sort Score
- Result 10 results
- Languages All
Results 31 - 40 of 1,897 for Devices (0.15 sec)
-
tensorflow/compiler/mlir/tensorflow/ir/tf_device.cc
Builder* builder, OperationState* state, int n, std::optional<DictionaryAttr> devices, llvm::ArrayRef<std::pair<ValueRange, Type>> replicated_inputs, ValueRange packed_inputs, TypeRange replica_output_types) { DCHECK_GE(n, 2); state->addAttribute("n", builder->getI32IntegerAttr(n)); if (devices.has_value()) state->addAttribute("devices", devices.value()); Region* region = state->addRegion();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 33.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/tpu_rewrite_pass.cc
// TODO(jpienaar): Remove this later. if (auto device = res->getAttrOfType<StringAttr>("device")) { if (!device.getValue().empty()) result_id->setAttr("device", device); else result_id->setAttr("device", compile_device_op); } else if (compile_device_op) { result_id->setAttr("device", compile_device_op); }
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/tf2xla/internal/passes/extract_head_tail_outside_compilation.cc
auto& side_effect_analysis = getAnalysis<mlir::TF::SideEffectAnalysis>(); // Get runtime devices information from the closest parent module. auto module = getOperation(); mlir::TF::RuntimeDevices devices; if (failed(tensorflow::GetDevicesFromOp(module, &devices))) return signalPassFailure(); OpBuilder builder(&getContext()); llvm::SmallVector<mlir::tf_device::ClusterOp, 4> clusters;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 20.6K bytes - Viewed (0) -
tensorflow/compiler/jit/flags.h
// `enabled_for_gpu_` below. bool enabled_for_all_; // If true, enable Device API (PjRt) for TF GPU device. This is a helper // flag so that individual tests can turn on PjRt for GPU specifically. // Once the rollout to GPU is complete, this flag can be deprecated. bool enabled_for_gpu_; private: // Devices for which using Device API (PjRt) is allowed in the XlaLaunch op.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 17 18:52:57 UTC 2024 - 14.5K bytes - Viewed (0) -
tensorflow/c/eager/c_api_test_util.cc
const string dev_name(TF_DeviceListName(devices, i, status.get())); CHECK_EQ(TF_GetCode(status.get()), TF_OK) << TF_Message(status.get()); if (dev_type == device_type) { *device_name = dev_name; LOG(INFO) << "Found " << device_type << " device " << *device_name; TF_DeleteDeviceList(devices); return true; } } TF_DeleteDeviceList(devices); return false; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 22:37:46 UTC 2024 - 23.5K bytes - Viewed (0) -
tensorflow/c/eager/c_api_experimental.h
// // device_info is an opaque caller-defined type stored with the custom device // which is passed to the functions referenced in the TFE_CustomDevice struct // `device` (execute, delete_device, etc.). It can for example contain the // names of wrapped devices. // // There are currently no graph semantics implemented for registered custom // devices, so executing tf.functions which contain operations placed on the
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 22:37:46 UTC 2024 - 39.5K bytes - Viewed (0) -
pkg/kubelet/apis/podresources/server_v1.go
// with device information provided by the DevicesProvider func NewV1PodResourcesServer(providers PodResourcesProviders) podresourcesv1.PodResourcesListerServer { return &v1PodResourcesServer{ podsProvider: providers.Pods, devicesProvider: providers.Devices, cpusProvider: providers.Cpus, memoryProvider: providers.Memory,
Registered: Sat Jun 15 01:39:40 UTC 2024 - Last Modified: Wed Feb 21 13:00:09 UTC 2024 - 6.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/convert_control_to_data_outputs.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 28.7K bytes - Viewed (0) -
tensorflow/c/experimental/saved_model/core/saved_variable_loading_test.cc
SessionOptions options; options.config.mutable_device_count()->insert({"CPU", 3}); std::vector<std::unique_ptr<Device>> devices; TF_CHECK_OK(DeviceFactory::AddDevices( options, "/job:localhost/replica:0/task:0", &devices)); device_mgr_ = absl::make_unique<StaticDeviceMgr>(std::move(devices)); ctx_ = testing::CreateTestingEagerContext(device_mgr_.get()); } EagerContext* context() { return ctx_.get(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Feb 27 09:34:33 UTC 2024 - 6.2K bytes - Viewed (0) -
pkg/kubelet/cm/dra/manager.go
annotations = append(annotations, claimAnnotations...) devices := claimInfo.cdiDevicesAsList() klog.V(3).InfoS("Add CDI devices", "claim", *claimName, "CDI devices", devices) cdiDevices = append(cdiDevices, devices...) return nil }) if err != nil { return nil, fmt.Errorf("locked cache operation: %w", err) } } }
Registered: Sat Jun 15 01:39:40 UTC 2024 - Last Modified: Fri May 03 13:23:29 UTC 2024 - 19.4K bytes - Viewed (0)