Searched defs:tf_context (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | meta_support.cc | 89 int GetWorkersCount(OpKernelContext* tf_context) { in GetWorkersCount()
118 void QuantizedGemmImpl(OpKernelContext* tf_context, const quint8* a_data, in QuantizedGemmImpl()
162 void MultiThreadTransform1D(OpKernelContext* tf_context, const Params& params) { in MultiThreadTransform1D()
220 void QuantizedGemm(OpKernelContext* tf_context, bool transpose_a, in QuantizedGemm()
256 void Requantize(OpKernelContext* tf_context, const qint32* input, int count, in Requantize()
295 void Dequantize(OpKernelContext* tf_context, const quint8* input, int count, in Dequantize()
319 void Quantize(OpKernelContext* tf_context, const float* input, int count, in Quantize()
351 void QuantizedBiasAdd(OpKernelContext* tf_context, const quint8* input, in QuantizedBiasAdd()
390 void Clamp(OpKernelContext* tf_context, const quint8* input, int count, in Clamp()
|
| /aosp_15_r20/external/mesa3d/src/gallium/frontends/teflon/ |
| H A D | tfl_device.c | 97 fill_operation(struct teflon_delegate *delegate, TfLiteContext *tf_context, TfLiteNode *node, TfLit… in fill_operation()
130 fill_tensor(struct teflon_delegate *delegate, TfLiteContext *tf_context, struct pipe_tensor *tensor… in fill_tensor()
220 partition_init(TfLiteContext *tf_context, const char *buffer, size_t length) in partition_init()
301 partition_free(TfLiteContext *tf_context, void *buffer) in partition_free()
314 partition_invoke(TfLiteContext *tf_context, TfLiteNode *node) in partition_invoke()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/flex/ |
| H A D | kernel.cc | 426 tensorflow::OpKernelContext tf_context(&run_state->params, in ExecuteOpKernelRunner() local
|