Home
last modified time | relevance | path

Searched defs:OpKernel (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/
H A Dlist_flex_ops.h27 struct OpKernel { struct
32 // The comparison function for OpKernel. argument
/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/
H A Dxla_kernel_creator.h25 class OpKernel; variable
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/shim/test_op/
H A Dsimple_tflite_op.cc25 using OpKernel = ::tflite::shim::TfLiteOpKernel<tflite::shim::SimpleOp>; typedef
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dlocal_executor_params.h29 class OpKernel; variable
H A Dgraph_view.h35 class OpKernel; variable
H A Dfunction.cc261 const std::shared_ptr<const NodeProperties>&, OpKernel**) { in CreateKernel() argument
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dop_kernel.cc130 OpKernel::OpKernel(OpKernelConstruction* context) : OpKernel(context, false) {} in OpKernel() function in tensorflow::OpKernel
132 OpKernel::OpKernel(OpKernelConstruction* context, bool is_deferred) in OpKernel() function in tensorflow::OpKernel
161 OpKernel::OpKernel(OpKernelConstruction* context, NodeDef&& custom_def, in OpKernel() function in tensorflow::OpKernel
H A Dfunction.h57 class OpKernel; variable
H A Dop_kernel.h355 friend class OpKernel; variable