Searched defs:split_input (Results 1 – 4 of 4) sorted by relevance
21 split_input = True variable in ElementBench
170 split_input = True variable in BroadcastBench
112 int split_input(int parallelism) { in split_input() function
319 HloInstruction* zero, bool split_input) { in GenerateBinaryMask()