Searched refs:inputTensorStrides (Results 1 – 1 of 1) sorted by relevance
1038 std::vector<ExprHandle> inputTensorStrides; in getInputStrides() local1042 std::vector<ExprHandle> inputTensorStrides; in getInputStrides() local1044 inputTensorStrides.push_back(LongImm::make(stride)); in getInputStrides()1046 return inputTensorStrides; in getInputStrides()1061 inputTensorStrides.resize(rank); in getInputStrides()1070 inputTensorStrides[i] = LongImm::make(1); in getInputStrides()1075 inputTensorStrides[i] = getStrideArg(input_index, i); in getInputStrides()1085 inputTensorStrides[i] = in getInputStrides()1086 inputTensorStrides[i + 1] * inputTensorDims[i + 1]; in getInputStrides()1095 inputTensorStrides[i] = in getInputStrides()[all …]