- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for XlaOpKernel (0.36 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/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)