- Sort Score
- Result 10 results
- Languages All
Results 21 - 26 of 26 for set_device (0.15 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/tpu_variable_runtime_reformatting.cc
execute_launch.getLoc(), llvm::ArrayRef<Type>{}, reformat_operands); WrapOpInLaunch(&builder, execute_launch.getLoc(), reformat_op, execute_launch.getDevice()); // Build the replicated unformat op after the loop. First prepare building the // replicate op. llvm::SmallVector<std::pair<ValueRange, Type>, 8> unformat_replicate_operands;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 21.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/mlrt/tpu_conversions.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Oct 04 21:25:31 UTC 2023 - 11K bytes - Viewed (0) -
pkg/volume/local/local.go
*localVolume readOnly bool } var _ volume.BlockVolumeMapper = &localVolumeMapper{} var _ volume.CustomBlockVolumeMapper = &localVolumeMapper{} // SetUpDevice prepares the volume to the node by the plugin specific way. func (m *localVolumeMapper) SetUpDevice() (string, error) { return "", nil } // MapPodDevice provides physical device path for the local PV. func (m *localVolumeMapper) MapPodDevice() (string, error) {
Registered: Sat Jun 15 01:39:40 UTC 2024 - Last Modified: Tue May 14 06:17:25 UTC 2024 - 22.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_dynamic_layout_pass.cc
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/transforms/replicate_to_island.cc
// Map aliased devices to explicit devices based on replica. if (auto launch = dyn_cast<tf_device::LaunchOp>(op)) if (auto device_by_replica = devices.value().get(launch.getDevice())) launch->setAttr( kDeviceAttr, device_by_replica.cast<ArrayAttr>()[replica_id].cast<StringAttr>()); return WalkResult::advance(); });
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Jul 24 21:01:40 UTC 2023 - 16.9K bytes - Viewed (0) -
pkg/kubelet/kubelet_volumes_test.go
return "", nil } func (f *stubBlockVolume) GetPodDeviceMapPath() (string, string) { return f.dirPath, f.volName } func (f *stubBlockVolume) SetUpDevice() (string, error) { return "", nil } func (f stubBlockVolume) MapPodDevice() error { return nil } func (f *stubBlockVolume) TearDownDevice(mapPath string, devicePath string) error { return nil
Registered: Sat Jun 15 01:39:40 UTC 2024 - Last Modified: Mon Apr 17 16:53:28 UTC 2023 - 20.4K bytes - Viewed (0)