- Sort Score
- Result 10 results
- Languages All
Results 11 - 16 of 16 for _device_ordinal (0.22 sec)
-
tensorflow/compiler/mlir/tensorflow/utils/device_util.cc
} mlir::LogicalResult GetDeviceOrdinalFromDeviceString(mlir::Location loc, llvm::StringRef device, int64_t* device_ordinal) { DeviceNameUtils::ParsedName parsed_name; if (!DeviceNameUtils::ParseFullName( absl::string_view(device.data(), device.size()), &parsed_name))
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/device_util.h
mlir::LogicalResult GetDeviceOrdinalFromDeviceString(mlir::Location loc, llvm::StringRef device, int64_t* device_ordinal); } // namespace tensorflow
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Aug 05 20:02:33 UTC 2020 - 2.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/replicate_to_island_legacy.mlir
// CHECK: tf_executor.fetch // Tests tf._TPUDeviceOrdinalPlaceholder ops are replaced with explicit device // ordinal constant values based on the first TPU core device id. // CHECK-LABEL: func @device_ordinals func.func @device_ordinals() { tf_executor.graph { %0:3 = tf_executor.island {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 31 08:59:10 UTC 2023 - 9.3K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_cpu_device.cc
XlaDevice::Options options; options.platform = platform; options.device_name_prefix = name_prefix; options.device_name = DEVICE_XLA_CPU; options.device_ordinal = 0; options.compilation_device_name = DEVICE_CPU_XLA_JIT; options.use_multiple_streams = false; XlaShapeLayoutHelpers::ShapeDeterminationFns shape_representation_fns{
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 5.5K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_gpu_device.cc
} for (int i : *gpu_ids) { XlaDevice::Options options; options.platform = platform.value(); options.device_name_prefix = name_prefix; options.device_name = DEVICE_XLA_GPU; options.device_ordinal = i; options.compilation_device_name = DEVICE_GPU_XLA_JIT; options.use_multiple_streams = true; options.allowed_devices = gpu_ids; XlaShapeLayoutHelpers::ShapeDeterminationFns shape_representation_fns{
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 6.6K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_tensor.h
// is replaced and the managed memory deallocated. Status AllocateShapedBuffer(DataType dtype, const xla::Shape& on_device_shape, xla::LocalClient* client, int device_ordinal); // Some Tensors can have complex on-device shapes, including tuple shapes. To // manage the memory for these tensors a ShapedBuffer may be required. // Return true if this XlaTensor contains a ShapedBuffer.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 4.7K bytes - Viewed (0)