- Sort Score
- Result 10 results
- Languages All
Results 71 - 80 of 847 for device_1 (0.27 sec)
-
tensorflow/compiler/mlir/tensorflow/tests/replicate_invariant_op_hoisting.mlir
%0:8 = tf_device.replicate([%arg0, %arg1] as %ri: tensor<*xf32>) {devices = {TPU_REPLICATED_CORE_0 = ["/device:TPU:0", "/device:TPU:1"]}, n = 2: i32} { %1 = "tf.Shape"(%ri) {device = "", T = "tfdtype$DT_FLOAT", out_type = "tfdtype$DT_INT32"} : (tensor<*xf32>) -> tensor<?xi32> %2 = "tf.opA"(%1) {device = "TPU_REPLICATED_CORE_0"} : (tensor<?xi32>) -> tensor<*xi32> %3 = "tf_device.launch"() ({
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 31 08:59:10 UTC 2023 - 11.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/tf_device_ops_invalid.mlir
// Check that replicate op's 'devices' attribute must consist of dictionary // with values as list with size equal to 'n' attribute. func.func @verifier_replicate_n_device_multiple_alias() { "tf_device.replicate" () ({ // expected-error@-1 {{'tf_device.replicate' op expects number of devices (2) to be equal to 'n' (3)}} ^entry: tf_device.return
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Aug 14 15:35:49 UTC 2023 - 9.8K bytes - Viewed (0) -
pkg/kubelet/cm/dra/claiminfo.go
func (info *ClaimInfo) cdiDevicesAsList() []kubecontainer.CDIDevice { var cdiDevices []kubecontainer.CDIDevice for _, devices := range info.CDIDevices { for _, device := range devices { cdiDevices = append(cdiDevices, kubecontainer.CDIDevice{Name: device}) } } return cdiDevices } // addPodReference adds a pod reference to the claim info.
Registered: Sat Jun 15 01:39:40 UTC 2024 - Last Modified: Fri May 03 13:30:31 UTC 2024 - 7.9K bytes - Viewed (0) -
tensorflow/c/experimental/saved_model/core/revived_types/restored_resource.h
// To keep things simple, I've stuck to raw pointers for now. // // Params: // device - The device string associated with the SavedResource // https://github.com/tensorflow/tensorflow/blob/master/tensorflow/core/protobuf/saved_object_graph.proto#L182 // Conceptually, this is the same device used in CapturableResource:
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 23 04:49:47 UTC 2020 - 4.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/tpu-dynamic-layout-pass.mlir
// CHECK-DAG: %[[COPY1:.*]] = "tf.TPUCopyWithLayout"(%[[ITER0]]#1, %[[LAYOUT1]]) {device = "/device:TPU:0"} // CHECK-DAG: %[[COPY2:.*]] = "tf.TPUCopyWithLayout"(%[[ITER1]]#0, %[[LAYOUT0]]) {device = "/device:TPU:1"} // CHECK-DAG: %[[COPY3:.*]] = "tf.TPUCopyWithLayout"(%[[ITER1]]#1, %[[LAYOUT1]]) {device = "/device:TPU:1"}
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 31 08:59:10 UTC 2023 - 29.6K bytes - Viewed (0) -
tensorflow/compiler/jit/flags.cc
"Switch a device into 'on-demand' mode, where instead of " "autoclustering ops are compiled one by one just-in-time."), Flag("tf_xla_enable_xla_devices", &device_flags->tf_xla_enable_xla_devices, "Generate XLA_* devices, where placing a computation on such a " "device" "forces compilation by XLA. Deprecated."),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 17 18:52:57 UTC 2024 - 24.5K 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_bridge_v1/end_to_end.mlir
%outputs_11, %control_12 = tf_executor.island wraps "tf.Placeholder"() {device = "", dtype = "tfdtype$DT_FLOAT", name = "y", shape = "tfshape$dim { }"} : () -> tensor<0xf32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 13 21:23:47 UTC 2024 - 3.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops.cc
auto node_device = [&](Operation *n) -> StringAttr { auto device = n->getAttrOfType<StringAttr>(device_id); if (!device || device.getValue().empty()) return caller_device; DeviceNameUtils::ParsedName ndef_parsed_device; if (!DeviceNameUtils::ParseFullName(device.getValue().str(), &ndef_parsed_device)) return device; DeviceNameUtils::MergeUnsetDevNames(&ndef_parsed_device,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 14.6K bytes - Viewed (0)