- Sort Score
- Result 10 results
- Languages All
Results 11 - 19 of 19 for _XlaCompile (0.14 sec)
-
tensorflow/compiler/jit/build_xla_ops_pass.cc
.NewSubScope(n->name()) .WithDevice(n->requested_device()) .WithAssignedDevice(device_name_str); ops::_XlaCompile xla_compile(root.WithOpName("xla_compile"), /*constants=*/cluster_info.constant_inputs, /*args=*/cluster_info.non_constant_inputs,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 24.3K bytes - Viewed (0) -
tensorflow/compiler/jit/tests/device_compiler_test_helper.cc
{{"f"}, "Add", {"e", "a"}, {{"T", DT_FLOAT}}}, {{"g"}, "Mul", {"f", "b"}, {{"T", DT_FLOAT}}}, // Force two clusters by excluding this node explicitly. {{"h"}, "Add", {"g", "f"}, {{"T", DT_FLOAT}, {"_XlaCompile", false}}}, {{"i"}, "Add", {"h", "e"}, {{"T", DT_FLOAT}}}, {{"j"}, "Add", {"i", "h"}, {{"T", DT_FLOAT}}}, {{"k"}, "Add", {"j", "h"}, {{"T", DT_FLOAT}}}, {{"l"}, "Add", {"k", "h"}, {{"T", DT_FLOAT}}},
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Feb 09 08:24:16 UTC 2024 - 6.2K bytes - Viewed (0) -
tensorflow/compiler/jit/flags.h
}; // Flags common to the _Xla* ops and their kernels. struct XlaOpsCommonFlags { // If true, _XlaCompile always refuses to compile the cluster, which means the // XLA clusters always run in the TF executor. Defaults to false. bool tf_xla_always_defer_compilation; // If true, _XlaCompile compiles the cluster asynchronously with respect to // the main execution. The fallback path is taken while compilation happens.
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/xla_device_ops.h
.HostMemory("resources"), \ KERNEL); #define REGISTER_XLA_COMPILE_KERNEL(DEVICE, KERNEL, TYPES) \ REGISTER_KERNEL_BUILDER(Name("_XlaCompile") \ .Device(DEVICE) \ .HostMemory("constants") \
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Nov 23 19:28:25 UTC 2021 - 17.1K bytes - Viewed (0) -
tensorflow/compiler/jit/kernels/xla_ops.cc
.HostMemory("constants") .HostMemory("resources"), XlaLocalLaunchOp); REGISTER_KERNEL_BUILDER(Name("_XlaCompile").Device(DEVICE_CPU), XlaCompileOp); REGISTER_KERNEL_BUILDER(Name("_XlaCompile") .Device(DEVICE_GPU) .HostMemory("constants") .HostMemory("key")
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/jit/mark_for_compilation_pass.cc
namespace tensorflow { namespace { using DeadnessPredicate = DeadnessAnalysis::DeadnessPredicate; using jit::DeviceId; using jit::DeviceSet; // The clusters we create here are eventually lowered into an // _XlaCompile/_XlaRun pair with a TF executor "fallback" that uses the // PartitionedCall op to execute the cluster in the regular graph executor if // need be. PartitionedCall, however, reruns the entire TF graph optimization
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 12:19:41 UTC 2024 - 85.3K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_platform_info.cc
// operations). Such a cluster can fail compilation (in way that // MarkForCompilation could not have detected) if the CPU JIT is not // linked in. // // So bail out of _XlaCompile in this case, and let the executor handle // the situation for us. const Status& status = compiler_for_platform.status(); if (status.code() == error::NOT_FOUND) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 17:23:27 UTC 2024 - 17.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc
effects.emplace_back(MemoryEffects::Write::get(), ResourceEffects::_XlaRun::get()); // Conservatively mark resource handles as read and write, as without // analyzing _XlaCompile, there is not sufficient information to determine // effects on resources. for (Value value : getArgs()) { MarkResourceAsReadAndWrite(value, effects); } }
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
node and associated metadata.}]>:$key, Res<TF_BoolTensor, [{If the `must_compile` attr is false the _XlaCompile op can decide not to compile the clusters based on some profitability heuristics. In that case `compilation_successful` is false if _XlaCompile chose not to compile the cluster. If the `must_compile` attr is true then _XlaCompile always attempts to compile the cluster and
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 23:24:08 UTC 2024 - 793K bytes - Viewed (0)