Searched refs:GetSoftmaxKernelCode (Results 1 – 3 of 3) sorted by relevance
27 std::string GetSoftmaxKernelCode(const OperationDef& op_def) { in GetSoftmaxKernelCode() function78 op.code_ = GetSoftmaxKernelCode(definition); in CreateSoftmax()
29 code_ = GetSoftmaxKernelCode(definition_); in Softmax1x1()41 std::string Softmax1x1::GetSoftmaxKernelCode(const OperationDef& op_def) { in GetSoftmaxKernelCode() function in tflite::gpu::Softmax1x1
47 std::string GetSoftmaxKernelCode(const OperationDef& op_def);