- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for send_key (0.1 sec)
-
tensorflow/compiler/mlir/tf2xla/internal/passes/extract_outside_compilation.cc
manual_inputs.push_back(in_manual); } // Create the _XlaHostComputeMlirOp auto host_compute = builder.create<mlir::TF::_XlaHostComputeMlirOp>( loc, shard_output_types, manual_inputs, /*send_key=*/builder.getStringAttr(args_communication_key), /*recv_key=*/builder.getStringAttr(retvals_communication_key), /*host_mlir_module=*/builder.getStringAttr(serialized_func_module),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 21:25:12 UTC 2024 - 68.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops.td
StrArrayAttr:$ancestors, TF_ShapeAttrArray:$shapes, OptionalAttr<SymbolRefAttr>:$shape_inference_graph, StrAttr:$key, DefaultValuedStrAttr<StrAttr, "">:$send_key, DefaultValuedStrAttr<StrAttr, "">:$recv_key, DefaultValuedOptionalAttr<I64Attr, "1000000">:$cost_estimate_ns, DefaultValuedOptionalAttr<I64Attr, "0">:$tpu_core ); let results = (outs
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/g3doc/_includes/tf_passes.md
}, { %1 = "tf_device.cluster"() ( { %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: Wed Aug 02 02:26:39 UTC 2023 - 96.4K bytes - Viewed (0)