- Sort Score
- Result 10 results
- Languages All
Results 51 - 60 of 1,897 for Devices (0.27 sec)
-
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_a_m.cc
// For EnqueueTPUEmbedding ops the device ordinal corresponds to the resource // instance. We also take the `device` attribute into account in order to avoid // dependencies between ops with the same ordinal on different devices. // Helper function to get an absolute device string, combining device and // ordinal attribute values.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 146.7K bytes - Viewed (0) -
pkg/kubelet/cm/dra/state/state_checkpoint.go
PodUIDs sets.Set[string] // ResourceHandles is a list of opaque resource data for processing by a specific kubelet plugin ResourceHandles []resourcev1alpha2.ResourceHandle // CDIDevices is a map of DriverName --> CDI devices returned by the // GRPC API call NodePrepareResource CDIDevices map[string][]string } // ClaimInfoStateWithoutResourceHandles is an old implementation of the ClaimInfoState // TODO: remove in Beta
Registered: Sat Jun 15 01:39:40 UTC 2024 - Last Modified: Fri May 03 13:23:27 UTC 2024 - 4.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/layout_optimization.cc
func::FuncOp func = getOperation(); // Get runtime devices information from the closest parent module. RuntimeDevices devices; if (failed(::tensorflow::GetDevicesFromOp(func->getParentOfType<ModuleOp>(), &devices))) return signalPassFailure(); // If there is no runtime device information and data format is not explicitly // forced, there is nothing to do.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 19.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/xla_broadcast.mlir
// RUN: tf-opt %s -split-input-file -tf-xla-broadcast | FileCheck %s
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 13 18:52:07 UTC 2024 - 2.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/cannonicalize_ops_outside_compilation.mlir
%2:3 = "tf.UnpackHyp"(%arg1) {_tpu_replicate = "cluster_Fn", _xla_outside_compilation = "0", device = "", max_seq_length = 16 : i64} : (tensor<*x!tf_type.string>) -> (tensor<*xi32>, tensor<*xi32>, tensor<*xf32>) %3 = "tf.Reshape"(%2#2, %1) {_tpu_replicate = "cluster_Fn", _xla_outside_compilation = "0", device = ""} : (tensor<*xf32>, tensor<*xi32>) -> tensor<*xf32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 13 21:23:47 UTC 2024 - 2.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/tpu_tail_with_tobool_op.mlir
%1 = "tf.Rank"(%0) {_tpu_replicate = "cluster", device = ""} : (tensor<*xi1>) -> tensor<*xi32> %2 = "tf.Range"(%cst_0, %1, %cst_1) {_tpu_replicate = "cluster", _xla_outside_compilation = "0", device = ""} : (tensor<i32>, tensor<*xi32>, tensor<i32>) -> tensor<*xi32> %3 = "tf.All"(%0, %2) {_tpu_replicate = "cluster", _xla_outside_compilation = "0", device = "", keep_dims = false} : (tensor<*xi1>, tensor<*xi32>) -> tensor<*xi1>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 13 21:23:47 UTC 2024 - 2.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/tpu-multiple-while-body-func.mlir
"func.func"() ({ ^bb0(%arg0: tensor<i32>): %0 = "tf.Less"(%arg0, %arg0) {device = ""} : (tensor<i32>, tensor<i32>) -> tensor<*xi1> %1 = "tf.Less"(%arg0, %arg0) {device = ""} : (tensor<i32>, tensor<i32>) -> tensor<*xi1> %2 = "tf.LogicalAnd"(%0, %1) {device = ""} : (tensor<*xi1>, tensor<*xi1>) -> tensor<*xi1> "func.return"(%2) : (tensor<*xi1>) -> ()
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 13 21:23:47 UTC 2024 - 2.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/xla_sharding_util.cc
const auto sharding_type = sharding.type(); if (sharding_type == xla::OpSharding::OTHER) { for (const auto& device : sharding.tile_assignment_devices()) { CHECK(device >= 0 && device < input_mappings.size()); input_mappings[device].push_back(idx); } } else if (sharding_type == xla::OpSharding::REPLICATED) { for (auto& input : input_mappings) input.push_back(idx);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 22 21:28:13 UTC 2024 - 34K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/cluster_tf_ops_pass.cc
std::string GetHost(Operation *op) { std::string device = ""; if (StringAttr attr = op->getAttrOfType<StringAttr>(kDeviceAttr)) { device = attr.getValue().str(); } return GetHost(device); } // The device is considered to be on the localhost iff one of the following is // true: // 1) None of the job/replica/task is specified in the device name. // 2) The job/replica/task in the device name are explicitly specified as
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 13.4K bytes - Viewed (0) -
tensorflow/c/eager/c_api_experimental.cc
// in an initialized context. for (auto d = devices.begin(); d != devices.end();) { if (absl::StrContains(d->get()->name(), "CPU:0")) { d = devices.erase(d); } else { ++d; } } status->status = tensorflow::unwrap(ctx)->AddDevices(std::move(devices)); } void TFE_InsertConfigKeyValue(TFE_Context* ctx, const char* key, const char* value, TF_Status* status) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 11 23:52:39 UTC 2024 - 35.9K bytes - Viewed (0)