Searched refs:stride_arr (Results 1 – 2 of 2) sorted by relevance
230 bool IsValidStrideCompactRowMajorData(int64_t* shape_arr, int64_t* stride_arr, in IsValidStrideCompactRowMajorData() argument232 if (ndim >= 1 && stride_arr[ndim - 1] != 1) { in IsValidStrideCompactRowMajorData()236 if (stride_arr[i] != shape_arr[i + 1] * stride_arr[i + 1]) { in IsValidStrideCompactRowMajorData()284 std::vector<int64_t>* stride_arr = &tf_dlm_tensor_ctx->strides; in TFE_HandleToDLPack() local286 stride_arr->resize(ndim, 1); in TFE_HandleToDLPack()291 (*stride_arr)[i] = (*shape_arr)[i + 1] * (*stride_arr)[i + 1]; in TFE_HandleToDLPack()300 dlm_tensor->dl_tensor.strides = stride_arr->data(); in TFE_HandleToDLPack()
22 dlm_tensor->dl_tensor.strides = &(*stride_arr)[0];