- Sort Score
- Result 10 results
- Languages All
Results 31 - 40 of 1,177 for kDevice (0.1 sec)
-
tensorflow/compiler/mlir/tensorflow/tests/tpu-variable-runtime-reformatting.mlir
%arg2: !tf_res_md_f32 {tf.device = "/device:TPU:0"}, %arg3: !tf_res_md_f32 {tf.device = "/device:TPU:1"}) { %0 = "tf.Const"() {value = dense<100> : tensor<i32>} : () -> tensor<i32> // CHECK: %[[STATE0:.*]] = "tf.VarHandleOp"() // CHECK-SAME: device = "/device:TPU:0" // CHECK: %[[STATE1:.*]] = "tf.VarHandleOp"() // CHECK-SAME: device = "/device:TPU:1"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 31 08:59:10 UTC 2023 - 25.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/tf_to_corert/insert_fallback_tensor_copy.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 25 10:51:48 UTC 2022 - 5.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/tpu_rewrite_device_util.cc
task_and_device = {task, device}; } } return topology; } // Determine execution devices when topology and device assignment are defined. // With a topology device coordinate to task and device mapping, device // assignment device coordinates can then be mapped to task and device for TPU // devices. The device assignment array is also validated. //
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Jun 10 20:10:40 UTC 2024 - 32.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/tf_to_corert/func_attributes_multiple_callers.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Nov 17 20:57:36 UTC 2022 - 4.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tf_device_assignment.cc
// We assign default device to ops with device attribute that is empty. if (device_attr.getValue().empty()) { op->setAttr(kDeviceAttr, builder.getStringAttr(default_device_)); } } else if (op->getDialect() == tf) { // Assign default device to all ops in Tensorflow dialect that do not // have device attribute.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Oct 05 23:50:19 UTC 2022 - 4.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/fuse-tftext.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 460.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/experimental/tac/tests/get-op-cost.mlir
// CHECK: tac.cost = 1.000000e+03 %0 = "tfl.add"(%arg0, %arg1) {fused_activation_function = "RELU", tac.device = "CPU"} : (tensor<10x10x10xf32>, tensor<10xf32>) -> tensor<10x10x10xf32> // CHECK: tac.cost = 1.000000e+03 %1 = "tfl.mul"(%0, %arg1) {fused_activation_function = "RELU", tac.device = "CPU"} : (tensor<10x10x10xf32>, tensor<10xf32>) -> tensor<10x10x10xf32> func.return %1 : tensor<10x10x10xf32> }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 24 05:29:10 UTC 2022 - 5.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_device_helper.cc
if (!device_attr || device_attr.getValue().empty()) return true; DeviceNameUtils::ParsedName device; if (!DeviceNameUtils::ParseFullName(device_attr.getValue().str(), &device)) return false; // We can't use GPU if operation explicitly placed on non-GPU device. return !device.has_type || device.type == ::tensorflow::DEVICE_GPU; } } // namespace TF
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 21 08:41:18 UTC 2022 - 2.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/tpu_device_propagation.mlir
// CHECK-SAME: ({{%.+}}: tensor<i64> {tf.device = "/job:localhost/replica:0/task:0/device:TPU:0"}, {{%.+}}: tensor<i32> {tf.device = "/job:localhost/replica:0/task:0/device:TPU:0"}) // CHECK-SAME: -> (tensor<i64> {tf.device = "/job:localhost/replica:0/task:0/device:TPU:0"}, tensor<i32> {tf.device = "/job:localhost/replica:0/task:0/device:TPU:0"})
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Mar 28 12:06:33 UTC 2022 - 19K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/tests/convert_tpu_model_to_cpu.mlir
%cst = "tf.Const"() {device = "", value = dense_resource<__elided__> : tensor<2x3x3x2xbf16>} : () -> tensor<2x3x3x2xbf16> %cst_0 = "tf.Const"() {device = "", value = dense<[0, 3, 1, 2]> : tensor<4xi32>} : () -> tensor<4xi32> %cst_1 = "tf.Const"() {_tpu_replicate = "cluster", device = "", value = dense<[0, 2, 3, 1]> : tensor<4xi32>} : () -> tensor<4xi32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 4.3K bytes - Viewed (0)