- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for XlaOpKernel (0.25 sec)
-
tensorflow/compiler/mlir/tf2xla/transforms/legalization_op_config.cc
namespace mlir { namespace mhlo { namespace { // Returns ops that should use MLIR legalization. // All other ops not in this list should use XlaOpKernel. const llvm::DenseSet<mlir::TypeID>& MlirAlwaysOps() { // The static variable is a pointer in order to avoid destruction upon thread // termination. static const llvm::DenseSet<mlir::TypeID>* ops = new llvm::DenseSet<
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 04:08:35 UTC 2024 - 21.7K bytes - Viewed (0) -
tensorflow/compiler/jit/compilability_check_util_test.cc
constexpr char kUncompilableFunctionNodeTwoName[] = "n_d_uncompilable"; // A dummy OpKernel for testing. class DummyCompilableOp : public XlaOpKernel { public: explicit DummyCompilableOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} void Compile(XlaOpKernelContext* ctx) override { ctx->SetOutput(0, ctx->Input(0)); } }; // Register the DummyCompilableOp kernel for CPU.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Jun 10 12:32:39 UTC 2022 - 22.3K bytes - Viewed (0) -
tensorflow/compiler/jit/compilability_check_util.h
bool allow_control_trigger = false; // Whether it is okay to "cluster" Assert and CheckNumerics by simply // removing them (they're not removed during clustering, but their // XlaOpKernel is a no-op kernel). We avoid auto-clustering these ops so // that the user is not surprised when XLA is implicitly enabled. If the // user explicitly specifies to use XLA, it is fine to resort to a dummy
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 14.9K bytes - Viewed (0)