Searched refs:dim2_copy_size (Results 1 – 2 of 2) sorted by relevance
29 int dim2_copy_size) { in CopyArrayData() argument33 for (int j = 0; j < dim2_copy_size; j++) { in CopyArrayData()77 int dim2_copy_size = is_bias ? 1 : sub_array.shape().dims(1); in CopyArrayToSubArray() local81 CopyArrayData(sub_buffer, dim2_copy_size, 0, 0, &tensor_buffer, in CopyArrayToSubArray()83 dim1_copy_size, dim2_copy_size); in CopyArrayToSubArray()
90 int dim2_copy_size);