Searched defs:index_array (Results 1 – 4 of 4) sorted by relevance
73 std::vector<HloInstruction*> index_array; in Run() local
623 glw::GLuint index_array = depth_position_in_array + y_position_in_array + x_position_in_array; in fillStencilData() local
593 def batch_shuffle(index_array, batch_size): argument
1771 index_array, argument