Searched refs:transpose_array (Results 1 – 2 of 2) sorted by relevance
220 auto& transpose_array = model->GetOrCreateArray(transpose_op->inputs[1]); in Run() local221 transpose_array.data_type = ArrayDataType::kInt32; in Run()222 transpose_array.GetMutableBuffer<ArrayDataType::kInt32>().data = new_perm; in Run()223 *(transpose_array.mutable_shape()->mutable_dims()) = { in Run()
193 auto& transpose_array = model->GetOrCreateArray(transpose_op->outputs[0]); in TransposeInput() local194 *transpose_array.mutable_shape()->mutable_dims() = in TransposeInput()