- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for GetDeviceOrdinal (0.27 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/replicate_to_island.cc
// `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. LogicalResult GetDeviceOrdinal(const std::optional<DictionaryAttr>& devices, const unsigned replica_id, const uint64_t logical_core,
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/xla_launch_util.h
xla::ExecuteOptions GetPjRtExecuteOptions( const DeviceType& device_type, absl::flat_hash_set<int> non_donatable_input_indices); // Returns the device ordinal from the parsed name of the device. int GetDeviceOrdinal(const DeviceBase* device); // Returns the device type from the OpKernelContext. DeviceType GetDeviceType(OpKernelContext* ctx); // Runs `executable` and populates the outputs in `ctx`. `inputs` and
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 09:53:30 UTC 2024 - 11.8K bytes - Viewed (0)