Searched refs:XlaCompileOnDemandOp (Results 1 – 4 of 4) sorted by relevance
31 XlaCompileOnDemandOp); \34 XlaCompileOnDemandOp); \36 XlaCompileOnDemandOp); \38 XlaCompileOnDemandOp); \40 XlaCompileOnDemandOp); \43 XlaCompileOnDemandOp); \45 XlaCompileOnDemandOp); \46 REGISTER_KERNEL_BUILDER(Name("XlaIf").Device(DEVICE), XlaCompileOnDemandOp); \52 XlaCompileOnDemandOp); \54 XlaCompileOnDemandOp); \[all …]
37 class XlaCompileOnDemandOp : public OpKernel {39 explicit XlaCompileOnDemandOp(OpKernelConstruction* ctx) in XlaCompileOnDemandOp() function
44 Status XlaCompileOnDemandOp::Run(OpKernelContext* ctx, in Run()102 Status XlaCompileOnDemandOp::Compile( in Compile()165 void XlaCompileOnDemandOp::Compute(OpKernelContext* ctx) { in Compute()
577 return new XlaCompileOnDemandOp(context); in RegisterXlaDeviceKernels()