Searched refs:shape_arr (Results 1 – 4 of 4) sorted by relevance
230 bool IsValidStrideCompactRowMajorData(int64_t* shape_arr, int64_t* stride_arr, in IsValidStrideCompactRowMajorData() argument236 if (stride_arr[i] != shape_arr[i + 1] * stride_arr[i + 1]) { in IsValidStrideCompactRowMajorData()283 std::vector<int64_t>* shape_arr = &tf_dlm_tensor_ctx->shape; in TFE_HandleToDLPack() local285 shape_arr->resize(ndim); in TFE_HandleToDLPack()288 (*shape_arr)[i] = tensor->dim_size(i); in TFE_HandleToDLPack()291 (*stride_arr)[i] = (*shape_arr)[i + 1] * (*stride_arr)[i + 1]; in TFE_HandleToDLPack()294 dlm_tensor->dl_tensor.shape = shape_arr->data(); in TFE_HandleToDLPack()
21 dlm_tensor->dl_tensor.shape = &(*shape_arr)[0];
712 SmallVector<int32_t, 8> shape_arr; in matchAndRewrite() local714 shape_arr.emplace_back(input_shape[i]); in matchAndRewrite()718 {static_cast<int32_t>(shape_arr.size())}, rewriter.getIntegerType(32)); in matchAndRewrite()720 shape_type, llvm::makeArrayRef<int32_t>(shape_arr)); in matchAndRewrite()
1630 SmallVector<int32_t, 8> shape_arr; in matchAndRewrite() local1632 shape_arr.emplace_back(input_shape[i]); in matchAndRewrite()1636 {static_cast<int32_t>(shape_arr.size())}, rewriter.getIntegerType(32)); in matchAndRewrite()1638 shape_type, llvm::makeArrayRef<int32_t>(shape_arr)); in matchAndRewrite()