- Sort Score
- Result 10 results
- Languages All
Results 11 - 15 of 15 for _device_ordinal (0.22 sec)
-
tensorflow/compiler/mlir/tf2xla/internal/mlir_bridge_pass_util_test.cc
Output x = ops::Placeholder(root.WithOpName("x"), DT_FLOAT); NameAttrList f_name_attr; f_name_attr.set_name("XTimesTwoFloat"); ops::TPUPartitionedCall f(root.WithOpName("f"), {x}, /*device_ordinal=*/0, {DT_FLOAT}, f_name_attr); TF_ASSERT_OK(root.ToGraph(&graph)); EXPECT_TRUE(IsInferenceGraph(graph, /*function_library=*/nullptr)); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 17 19:51:50 UTC 2024 - 10.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops.td
ConfinedAttr<TypeArrayAttr, [ArrayMinCount<1>]>:$dtypes, TF_ShapeAttrArray:$shapes, DefaultValuedOptionalAttr<I64ArrayAttr, "{}">:$layouts, DefaultValuedOptionalAttr<I64Attr, "-1">:$device_ordinal ); let results = (outs); } // This op is manually defined because the attribute name `template` (which is // a keyword) is changed to `strtemplate`.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 04:08:35 UTC 2024 - 90.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_a_m.cc
// Helper function to get an absolute device string, combining device and // ordinal attribute values. std::string GetAbsDeviceStr(Operation* op, uint64_t device_ordinal) { std::string device_ordinal_str = std::to_string(device_ordinal); auto device_attr = op->getAttrOfType<StringAttr>("device"); if (!device_attr || device_attr.getValue().empty()) return device_ordinal_str;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 146.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td
which to execute computations, such as `TPUPartitionedCall`. device_ordinal: An integer in [0, NumTPUCoresVisiblePerHost]. }]; let arguments = (ins); let results = (outs TF_Int32Tensor:$device_ordinal ); } def TF_TakeDatasetOp : TF_Op<"TakeDataset", [Pure]> { let summary = [{
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 23:24:08 UTC 2024 - 793K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/tpu_rewrite.mlir
%1 = "tf_device.launch"() ({ %2 = "tf._XlaCompileMlirPlaceholderProgramKey"() : () -> tensor<3x!tf_type.string> %3 = "tf._XlaRecvAtHost"(%2) {_xla_has_host_transfer = true, device_ordinal = 0 : i64, key = "host_compute_channel_0_args"} : (tensor<3x!tf_type.string>) -> tensor<?xi32> %4 = "tf.B"(%3) : (tensor<?xi32>) -> tensor<?x!tf_type.string> tf_device.return %4 : tensor<?x!tf_type.string>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 22:03:30 UTC 2024 - 172.9K bytes - Viewed (0)