- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for XlaOpKernel (0.18 sec)
-
tensorflow/compiler/aot/aot_only_var_handle_op.cc
// same name. It is not safe to use this op in a JIT context. class XlaAotOnlyVarHandleOp : public XlaOpKernel { public: explicit XlaAotOnlyVarHandleOp(OpKernelConstruction* c); void Compile(XlaOpKernelContext* context) override; private: string name_; }; XlaAotOnlyVarHandleOp::XlaAotOnlyVarHandleOp(OpKernelConstruction* c) : XlaOpKernel(c) { OP_REQUIRES_OK(c, c->GetAttr("shared_name", &name_)); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 28 09:57:04 UTC 2024 - 3K bytes - Viewed (0) -
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/xla_cluster_util.h
// The attribute that marks certain inputs to a Node as required to be a // constant at compile time. If this attribute is present then the // CompileTimeConstantInput information in the corresponding XlaOpKernel is // ignored. // // The value for this attribute, if present, has to be a list of strings naming // the inputs to the node that must be constant. extern const char* const kXlaCompileTimeConstantInputsAttr;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 09:53:30 UTC 2024 - 4.6K bytes - Viewed (0)