Searched refs:axis_to_selector (Results 1 – 2 of 2) sorted by relevance
36 std::map<Axis, std::string> axis_to_selector = { in GetConcatKernelCode() local95 "args." + tensor_names[i] + "." + axis_to_selector[attr.axis] + "()"; in GetConcatKernelCode()
252 std::map<Axis, std::string> axis_to_selector = { in GetReduceKernelCode() local267 axis_to_selector[axis]; in GetReduceKernelCode()313 " < args.src_tensor." + axis_to_selector[axis] + "; " + src_coord + in GetReduceKernelCode()