Home
last modified time | relevance | path

Searched defs:XlaOpKernelContext (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/
H A Dxla_op_kernel.h32 class XlaOpKernelContext; variable
H A Dxla_context.h36 class XlaOpKernelContext; variable
H A Dxla_op_kernel.cc37 XlaOpKernelContext::XlaOpKernelContext(OpKernelContext* context) in XlaOpKernelContext() function in tensorflow::XlaOpKernelContext
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dop_kernel.h1682 class XlaOpKernelContext; variable
1683 inline void CheckNotInComputeAsync(XlaOpKernelContext*, const char*) {} in CheckNotInComputeAsync() argument