Searched refs:partition_array (Results 1 – 2 of 2) sorted by relevance
198 const auto& partition_array = model->GetArray(gather_ops[0]->inputs[0]); in Run() local199 const auto& partition_array_dims = partition_array.shape().dims(); in Run()201 partition_array.shape().dimensions_count(); in Run()220 merged_gather_op->input_rank = partition_array.shape().dimensions_count(); in Run()
148 partition_array = io.BytesIO()150 part.generate(partition_array)151 self.header.part_entries_crc32 = zlib.crc32(partition_array.getvalue())