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