- Sort Score
- Result 10 results
- Languages All
Results 1 - 9 of 9 for XlaLaunch (0.37 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/tf_device_passes.td
This pass rewrites `tf.PartitionedCall` and `tf.StatefulPartitionedCall` operations with `_xla_compile_device_type` attribute in a `tf_device.cluster` into `tf.XlaLaunch` operations. This makes the attached function execute with XLA. `tf.XlaLaunch` requires resource-type arguments come at the end, so this pass rewrites the called function if necessary. This pass assumes there are no nested `tf_device.cluster`s so we don't end
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 17 18:52:57 UTC 2024 - 12.5K bytes - Viewed (0) -
tensorflow/compiler/jit/flags.h
public: // Allow using Device API (PjRt) for `device_type` in the XlaLaunch op. // Please note that `enabled_for_xla_launch_` needs to be true in addition // to the `device_type` being allowed in order to use the Device API for // single device compilation and execution in the XlaLaunch op. void AllowForDeviceInXlaLaunch(const DeviceType& device_type) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 17 18:52:57 UTC 2024 - 14.5K bytes - Viewed (0) -
tensorflow/compiler/jit/kernels/xla_ops.cc
if (ctx->has_input(i) || ctx->has_input(++i)) { ctx->set_output(0, ctx->input(i)); } } REGISTER_KERNEL_BUILDER(Name("XlaLaunch").Device(DEVICE_CPU), XlaLocalLaunchOp); REGISTER_KERNEL_BUILDER(Name("XlaLaunchV2").Device(DEVICE_CPU), XlaLaunchV2Op); REGISTER_KERNEL_BUILDER(Name("XlaLaunch") .Device(DEVICE_GPU) .HostMemory("constants")
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 22:46:36 UTC 2024 - 41.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/lower_cluster_to_runtime_ops.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 17 18:52:57 UTC 2024 - 9.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/mlrt/tf_to_mlrt.mlir
%unused = "tf.TestAsyncIdentity"(%x) {__op_key = 0: i32, T = i32} : (tensor<i32>) -> tensor<i32> // CHECK: mlrt.await_all_control [[unused]] return %x : tensor<i32> } // ----- // Test for XlaLaunch func.func private @xla_func_0(%arg0: tensor<1x3xf32>, %arg1: tensor<1x3xf32>) -> tensor<1x3xf32> attributes {tf._XlaMustCompile = true, tf._noinline = true, tf._original_func_name = "should_not_be_used"} {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 31 20:44:15 UTC 2024 - 24.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/clustering_passes.td
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/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) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc
ResourceEffects::XlaLaunch::get()); // Conservatively mark resource handles as read and write, as without // analyzing XlaLaunch, there is not sufficient information to determine // effects on resources. for (Value value : getArgs()) { MarkResourceAsReadAndWrite(value, effects); } } // For `XlaLaunch` ops the `device` attribute corresponds to the resource
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 22:07:10 UTC 2024 - 170.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td
); TF_DerivedOperandTypeAttr K = TF_DerivedOperandTypeAttr<0>; TF_DerivedOperandTypeAttr V = TF_DerivedOperandTypeAttr<1>; } def TF_XlaLaunchOp : TF_Op<"XlaLaunch", [AttrSizedOperandSegments, DeclareOpInterfaceMethods<MemoryEffectsOpInterface>, DeclareOpInterfaceMethods<TF_GetResourceInstanceInterface>]> { let summary = "XLA Launch Op. For use by the XLA JIT only.";
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 23:24:08 UTC 2024 - 793K bytes - Viewed (0)