- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 43 for _device_ordinal (0.37 sec)
-
tensorflow/compiler/jit/xla_launch_util.cc
xla::LocalClient* client, se::DeviceMemoryAllocator* xla_allocator, int device_ordinal, bool allocate_xla_tensors, bool use_multiple_streams) : client_(client), xla_allocator_(xla_allocator), allocate_xla_tensors_(allocate_xla_tensors), use_multiple_streams_(use_multiple_streams), device_ordinal_(device_ordinal) { if (use_multiple_streams_) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 16 00:36:08 UTC 2024 - 40.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/replica_id_to_device_ordinal.cc
if (op->getAttrOfType<IntegerAttr>(kDeviceOrdinalAttr).getInt() == -1) { OpBuilder builder(op); int device_ordinal = op->getAttrOfType<IntegerAttr>(kReplicaIdAttr).getInt() % device_num; op->setAttr(kDeviceOrdinalAttr, builder.getI64IntegerAttr(device_ordinal)); } } } } } // namespace std::unique_ptr<OperationPass<func::FuncOp>>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Oct 05 23:50:19 UTC 2022 - 4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/tpu_update_embedding_enqueue_op_inputs.mlir
"tf.EnqueueTPUEmbeddingSparseTensorBatch"(%arg0, %arg1, %arg2, %arg3, %arg4, %arg5, %0, %0, %0, %arg7) {_tpu_embedding_layer = "call1", _xla_outside_compilation = "0", combiners = ["mean", "sum"], device_ordinal = -1 : i64, max_sequence_lengths = [0, 0, 0], table_ids = [1, 1, 0]} : (tensor<?x2xi32>, tensor<?x2xi32>, tensor<?x2xi32>, tensor<?xi32>, tensor<?xi32>, tensor<?xi32>, tensor<0xf32>, tensor<0xf32>, tensor<0xf32>, tensor<!tf_type.string>) -> ()...
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 5.3K bytes - Viewed (0) -
tensorflow/compiler/jit/pjrt_base_device.h
Options(std::string device_name_prefix, std::string device_name, int device_ordinal, std::string compilation_device_name, std::vector<XlaShapeLayoutHelpers::ShapeDeterminationFns> shape_determination_fns) : device_name_prefix(device_name_prefix), device_name(device_name), device_ordinal(device_ordinal), compilation_device_name(compilation_device_name),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 12:19:41 UTC 2024 - 4K bytes - Viewed (0) -
tensorflow/compiler/jit/pjrt_base_device.cc
const string& device_name, int device_ordinal) { return Device::BuildDeviceAttributes( absl::StrCat(name_prefix, "/device:", device_name, ":", device_ordinal), DeviceType(device_name), Bytes(16ULL << 30), DeviceLocality(), absl::StrCat("device: ", device_name, " device")); } } // namespace
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 12:19:41 UTC 2024 - 2.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/replicate_to_island.cc
} // Returns the device ordinal (`device_ordinal`) for a replica (`replica_id`) // and logical core (`logical_core`). // `replica_id` is the index of the ancestor ReplicateOp in [0, num_replicas). // `logical_core` is the index of the TPU core in [0, num_cores_per_replica). // `device_ordinal` is the index of the TPU core relative to its host.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Jul 24 21:01:40 UTC 2023 - 16.9K bytes - Viewed (0) -
tensorflow/compiler/jit/device_compiler_client.cc
xla::ExecutableBuildOptions build_options; if (result.collective_info) { build_options.set_num_replicas(result.collective_info->group_size); } if (options.device_ordinal != -1) { build_options.set_device_ordinal(options.device_ordinal); } else if (default_device_ordinal != -1) { build_options.set_device_ordinal(default_device_ordinal); } build_options.set_result_layout(result.xla_output_shape);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Oct 26 20:35:26 UTC 2023 - 1.8K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_tensor.cc
int device_ordinal) { xla::Shape on_host_shape = xla::ShapeUtil::DeviceShapeToHostShape(on_device_shape); xla::ScopedShapedBuffer shaped_buffer(on_host_shape, on_device_shape, client->backend().memory_allocator(), device_ordinal); for (auto& index_to_buffer : shaped_buffer.buffers()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 4.5K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_compiler_options_util_test.cc
layout_preference_fn, shape_representation_fn}}; } std::unique_ptr<XlaDevice::Metadata> CreateXlaDeviceMetadata( DeviceType compilation_device_type) { return std::make_unique<XlaDevice::Metadata>( /*device_ordinal=*/0, /*platform=*/nullptr, compilation_device_type, GetShapeDeterminationFns(), XlaDevice::PaddedShapeFn(), /*use_multiple_streams=*/false); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Dec 29 01:41:20 UTC 2023 - 14.8K bytes - Viewed (0) -
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)