- Sort Score
- Result 10 results
- Languages All
Results 21 - 30 of 155 for r2devices (0.14 sec)
-
tensorflow/c/eager/parallel_device/parallel_device_lib.h
// A parallel tensor with scalar integers numbering component devices. std::unique_ptr<ParallelTensor> DeviceIDs(TFE_Context* context, TF_Status* status) const; // The number of devices operations run on. size_t num_underlying_devices() const { return underlying_devices_.size(); } // The devices operations run on. const std::vector<std::string>& underlying_devices() const {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 25 15:21:13 UTC 2023 - 12.9K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_device.h
// If padded_shape_fn is empty, a default implementation that returns // the logical on-device shape without padding is used. PaddedShapeFn padded_shape_fn; // Set of devices to use. This controls which of the devices on the given // platform will have resources allocated. For GPUs this will be // filled from visible_gpu_devices list from session configuration. std::optional<std::set<int>> allowed_devices;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 09:53:30 UTC 2024 - 13.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/tpu_rewrite_pass.cc
return tensorflow::WrapOpInLaunch(builder, compile_op.getLoc(), compile_op, compilation_device); } // Assigns explicit devices to replicate op. An aliased device is created per // core, and all replica devices per core are grouped together. void AssignDevicesToReplicate( tf_device::ReplicateOp replicate, llvm::ArrayRef<llvm::SmallVector<tensorflow::TPUDeviceAndHost, 8>>
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) -
pkg/kubelet/cm/devicemanager/plugin/v1beta1/stub.go
klog.InfoS("ListAndWatch") s.Send(&pluginapi.ListAndWatchResponse{Devices: m.devs}) for { select { case <-m.stop: return nil case updated := <-m.update: s.Send(&pluginapi.ListAndWatchResponse{Devices: updated}) } } } // Update allows the device plugin to send new devices through ListAndWatch func (m *Stub) Update(devs []*pluginapi.Device) { m.update <- devs }
Registered: Sat Jun 15 01:39:40 UTC 2024 - Last Modified: Tue Oct 17 11:19:10 UTC 2023 - 11.5K bytes - Viewed (0) -
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/tests/tpu-variable-runtime-reformatting.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 31 08:59:10 UTC 2023 - 25.4K bytes - Viewed (0) -
tensorflow/c/eager/c_api_test_util.cc
TF_NewStatus(), TF_DeleteStatus); TF_DeviceList* devices = TFE_ContextListDevices(ctx, status.get()); CHECK_EQ(TF_OK, TF_GetCode(status.get())) << TF_Message(status.get()); const int num_devices = TF_DeviceListCount(devices); for (int i = 0; i < num_devices; ++i) { const string dev_type(TF_DeviceListType(devices, i, status.get()));
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
size_t proto_len, TF_Status* status); // TODO(b/166642410): It would be nice, for custom devices and for other users, // to have a non-string representation of devices (TF_Device) extracted from // tensors/ops/etc. and usable in APIs like OpSetDevice/ResetOp/etc. #define TFE_CUSTOM_DEVICE_VERSION 4
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/cm/dra/claiminfo_test.go
expectedResult []kubecontainer.CDIDevice }{ { description: "empty CDI devices", claimInfo: &ClaimInfo{ ClaimInfoState: state.ClaimInfoState{ CDIDevices: map[string][]string{}, }, }, }, { description: "nil CDI devices", claimInfo: &ClaimInfo{}, }, { description: "valid CDI devices", claimInfo: &ClaimInfo{ ClaimInfoState: state.ClaimInfoState{
Registered: Sat Jun 15 01:39:40 UTC 2024 - Last Modified: Fri May 03 13:30:31 UTC 2024 - 21K bytes - Viewed (0)