Searched defs:GetSoftmaxKernelCode (Results 1 – 2 of 2) sorted by relevance
27 std::string GetSoftmaxKernelCode(const OperationDef& op_def) { in GetSoftmaxKernelCode() function
41 std::string Softmax1x1::GetSoftmaxKernelCode(const OperationDef& op_def) { in GetSoftmaxKernelCode() function in tflite::gpu::Softmax1x1