- Sort Score
- Result 10 results
- Languages All
Results 21 - 26 of 26 for XlaLaunch (0.22 sec)
-
tensorflow/compiler/jit/compilability_check_util.h
const tensorflow::FunctionBody* fbody, absl::Span<int const> constant_arg_indices, absl::Span<int const> resource_arg_indices); // Returns output memory types. // // XlaLaunch kernel keeps all outputs (including constants, which it copies), // in device memory except for resources. tensorflow::MemoryTypeVector GetOutputMemoryTypes( const tensorflow::FunctionBody* fbody);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 14.9K 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/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) -
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)