Searched refs:GetReduceKernelCode (Results 1 – 2 of 2) sorted by relevance
50 std::string GetReduceKernelCode(const OperationDef& op_def,
144 code_ = GetReduceKernelCode(definition_, work_group_size_, in Reduce()160 std::string Reduce::GetReduceKernelCode(const OperationDef& op_def, in GetReduceKernelCode() function in tflite::gpu::Reduce