- Sort Score
- Result 10 results
- Languages All
Results 11 - 12 of 12 for XlaLaunch (0.19 sec)
-
tensorflow/compiler/mlir/tfrt/tests/hoist_invariant_ops.mlir
attributes {tf_saved_model.exported_names = ["main"]} { %0 = "tf.VarHandleOp"() {device = "/device:CPU:0", container = "", shared_name = "variable"} : () -> tensor<!tf_type.resource<tensor<1x3xf32>>> %1 = "tf.XlaLaunch"(%arg0, %0) {device = "/device:GPU:0", function = @xla_func, operandSegmentSizes = array<i32: 0, 2, 0>} : (tensor<1x3xf32>, tensor<!tf_type.resource<tensor<1x3xf32>>>) -> tensor<*xf32> func.return %1 : tensor<*xf32> }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 01 23:54:14 UTC 2024 - 18.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/passes.h
// parent region. std::unique_ptr<OperationPass<ModuleOp>> CreateXlaInlineDeviceOpsPass(); // Creates a pass that rewrites partitioned calls with `_xla_compile_device // type` with `tf.XlaLaunch` ops. std::unique_ptr<OperationPass<ModuleOp>> CreateXlaRewritePass(); // Create a pass that validates the input graph to the CPU/GPU bridge. std::unique_ptr<OperationPass<ModuleOp>> CreateXlaValidateInputsPass();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:18:05 UTC 2024 - 31.8K bytes - Viewed (0)