Searched defs:has_batch (Results 1 – 3 of 3) sorted by relevance
63 bool has_batch; member
93 const bool has_batch = data->shape().rank() == 2; in Run() local
2260 const bool has_batch = hlo->shape().tuple_shapes(0).dimensions_size() == 2; in HandleTopK() local