Searched refs:XlaCompileOnDemandOp (Results 1 – 3 of 3) sorted by relevance
50 Status XlaCompileOnDemandOp::Run(OpKernelContext* ctx, in Run()91 Status XlaCompileOnDemandOp::MustArgumentBeConstant(const OpKernel* op_kernel, in MustArgumentBeConstant()104 Status XlaCompileOnDemandOp::ShouldArgumentBeConstant(const OpKernel* op_kernel, in ShouldArgumentBeConstant()113 Status XlaCompileOnDemandOp::Compile( in Compile()211 void XlaCompileOnDemandOp::Compute(OpKernelContext* ctx) { in Compute()
34 class XlaCompileOnDemandOp : public OpKernel {36 explicit XlaCompileOnDemandOp(OpKernelConstruction* ctx) : OpKernel(ctx) {} in XlaCompileOnDemandOp() function
526 return new XlaCompileOnDemandOp(context); in RegisterXlaDeviceKernels()