Searched defs:op_kernel_ctx (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/ |
| H A D | tf_framework_c_interface.cc | 52 Allocator* GetAllocator(void* op_kernel_ctx) { in GetAllocator()
61 extern "C" void* _mlir_ciface_tf_alloc(void* op_kernel_ctx, size_t num_elements, in _mlir_ciface_tf_alloc()
96 extern "C" void _mlir_ciface_tf_dealloc(void* op_kernel_ctx, void* ptr) { in _mlir_ciface_tf_dealloc()
100 extern "C" void _mlir_ciface_tf_report_error(void* op_kernel_ctx, in _mlir_ciface_tf_report_error()
113 static void ReportError(void* op_kernel_ctx, ErrorCode error_code, in ReportError()
241 void* op_kernel_ctx, char* code, int64_t num_tile_sizes, in _mlir_ciface_tf_jit_compile()
295 extern "C" void _mlir_ciface_tf_jit_execute(void* op_kernel_ctx, void* callable, in _mlir_ciface_tf_jit_execute()
|
| H A D | tf_gpu_runtime_wrappers.cc | 119 auto *op_kernel_ctx = static_cast<tensorflow::OpKernelContext *>(ctx); in _mlir_ciface_tf_launch_kernel() local
|