- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 16 for tpu_core (0.13 sec)
-
tensorflow/compiler/mlir/tensorflow/tests/mlir2graphdef/optional_symbol_ref.mlir
tf_executor.fetch } func.return } } // CHECK: op: "XlaHostCompute" // CHECK: attr { // CHECK: key: "shape_inference_graph" // CHECK: value {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Mar 28 12:06:33 UTC 2022 - 967 bytes - Viewed (0) -
tensorflow/compiler/jit/encapsulate_subgraphs_pass_test.cc
{"key", "host_compute_channel_F1_F1_O1"}, {"send_key", ""}, {"recv_key", ""}, {"shape_inference_graph", shape_inference_graph}, {"tpu_core", 0}, {"cost_estimate_ns", 1000000}, {"shapes", absl::Span<const DataType>({})}, {"_outside_compilation_subgraph", "O1"}, {"_xla_token_input_nodes",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 113.3K bytes - Viewed (0) -
tensorflow/compiler/jit/extract_outside_compilation_pass_test.cc
EXPECT_NE(host_compute_1, nullptr); // Check XlaHostCompute nodes' "tpu_core" attr. int tpu_core; TF_CHECK_OK(GetNodeAttr(host_compute_0->attrs(), "tpu_core", &tpu_core)); EXPECT_EQ(tpu_core, 1); TF_CHECK_OK(GetNodeAttr(host_compute_1->attrs(), "tpu_core", &tpu_core)); EXPECT_EQ(tpu_core, 0); // Check XlaHostCompute nodes' "shapes" attr. "0" should not have shapes, and
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 41K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/prepare_tpu_computation_for_tf_export.mlir
// CHECK-SAME: send_key = "host_compute_channel_send" // CHECK-SAME: shape_inference_graph = @host_func // CHECK-SAME: shapes = [#tf_type.shape<*>, #tf_type.shape<3x?>] // CHECK-SAME: tpu_core = 0 : i64 // CHECK: func @host_func // CHECK: %[[RECV_OUTPUT:[0-9]*]]:2 = "tf._XlaRecvAtHost" // CHECK-SAME: key = "host_compute_channel_send" // CHECK-SAME: -> (tensor<*xf32>, tensor<3x?xf64>)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 14 18:46:36 UTC 2024 - 9.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/prepare_tpu_computation_for_tf_export.cc
/*key=*/rewriter.getStringAttr(""), op.getSendKeyAttr(), op.getRecvKeyAttr(), /*cost_estimate_ns=*/rewriter.getI64IntegerAttr(kDefaultCostEstimate), /*tpu_core=*/rewriter.getI64IntegerAttr(0)); } }; void UpdateArgAttributes(mlir::func::FuncOp func) { OpBuilder builder(func.getBody()); for (int i = 0; i < func.getNumArguments(); ++i) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/clustering_passes.td
%2 = "tf.Const"() {value = dense<1.000000e+00> : tensor<f32>} : () -> tensor<f32> %3 = "tf._XlaHostComputeMlir"(%2) {recv_key = "host_compute_channel_0_0_retvals", send_key = "host_compute_channel_0_0_args", tpu_core = 0 : i64} : (tensor<f32>) -> tensor<f32> %4 = "tf.AddV2"(%2, %3) : (tensor<f32>, tensor<f32>) -> tensor<f32> tf_device.return %4 : tensor<f32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 02:01:13 UTC 2024 - 19.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/mark_ops_for_outside_compilation.mlir
"tf.XlaHostCompute"(%cst) <{ancestors = [], cost_estimate_ns = 1000000 : i64, key = "_host_callback", recv_key = "", send_key = "", shapes = [], tpu_core = 0 : i64}> {_xla_original_oc_node_name = "hcb0", _xla_token_input_nodes = ["_xla_token_arg_node"]} : (tensor<i32>) -> () tf_device.return
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 16:22:32 UTC 2024 - 29.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/sparsecore/embedding_sequencing.cc
metadata_op, compilation_op, loop_body_func, "sc_forward", &forward_caller); if (failed(result)) return signalPassFailure(); // Create tpu_core function builder.setInsertionPointAfter(forward_caller); Operation* core_tpu_caller = nullptr; result = ExtractOpsAsFunc(builder, module, core_tpu_ops, replication_attr,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 39.4K bytes - Viewed (0) -
tensorflow/compiler/jit/extract_outside_compilation_pass.cc
for (const auto& attr : call_node->attrs()) { host_compute_builder.Attr(attr.first, attr.second); } // Populate tpu_core assignment. const auto iter = host_compute_core.find(original_oc_name); if (iter != host_compute_core.end()) { int core = iter->second; host_compute_builder.Attr("tpu_core", core); } // Set input tokens and other outside compilation clusters that current
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 104.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/shape_inference.mlir
// CHECK: "tf._XlaHostComputeMlir" // CHECK-SAME: -> tensor<*xf32> %0 = "tf._XlaHostComputeMlir"(%arg0) {recv_key = "host_compute_channel_recv", send_key = "host_compute_channel_send", tpu_core = 0, host_mlir_module = ""} : (tensor<2xf32>) -> tensor<*xf32> func.return %0 : tensor<*xf32> } // CHECK-LABEL: func @xla_host_compute_mlir_shape_inferred
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jan 23 17:24:10 UTC 2024 - 167.4K bytes - Viewed (0)