Home
last modified time | relevance | path

Searched defs:outputStrides (Results 1 – 14 of 14) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/
H A Dlowerings.cpp59 at::Device device) { in nnc_lowerings_lazy_registration()
110 const std::vector<ExprHandle>& outputStrides, \ in nnc_lowerings_lazy_registration()
157 const std::vector<ExprHandle>& outputStrides, \ in nnc_lowerings_lazy_registration()
247 const std::vector<ExprHandle>& outputStrides, in nnc_lowerings_lazy_registration()
266 const std::vector<ExprHandle>& outputStrides, in nnc_lowerings_lazy_registration()
285 const std::vector<ExprHandle>& outputStrides, in nnc_lowerings_lazy_registration()
304 const std::vector<ExprHandle>& outputStrides, in nnc_lowerings_lazy_registration()
323 const std::vector<ExprHandle>& outputStrides, in nnc_lowerings_lazy_registration()
342 const std::vector<ExprHandle>& outputStrides, in nnc_lowerings_lazy_registration()
361 const std::vector<ExprHandle>& outputStrides, in nnc_lowerings_lazy_registration()
[all …]
H A Dtensorexpr_init.cpp728 std::vector<ExprHandle> outputStrides = in initTensorExprBindings() local
H A Dkernel.cpp579 std::vector<ExprHandle> outputStrides = {}; in computeValue() local
/aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/operators/
H A Dquantization.cpp141 const std::vector<ExprHandle>& outputStrides, in computeQuantizePerTensor()
181 const std::vector<ExprHandle>& outputStrides, in computeQuantizedAdd()
224 const std::vector<ExprHandle>& outputStrides, in computeQuantizePerTensorExternalCall()
256 const std::vector<ExprHandle>& outputStrides, in computeDequantizeExternalCall()
281 const std::vector<ExprHandle>& outputStrides, in computeQuantizedConv2dPrepack()
324 const std::vector<ExprHandle>& outputStrides, in computeQuantizedConv1d()
354 const std::vector<ExprHandle>& outputStrides, in computeQuantizedConv2d()
384 const std::vector<ExprHandle>& outputStrides, in computeQuantizedConv2dRelu()
414 const std::vector<ExprHandle>& outputStrides, in computeQuantizedLinear()
444 const std::vector<ExprHandle>& outputStrides, in computeQuantizedLinearRelu()
[all …]
H A Dpointwise.cpp11 const std::optional<std::vector<ExprHandle>>& outputStrides) { in computeSign()
28 const std::vector<ExprHandle>& outputStrides, in computeOneOperand()
51 const std::vector<ExprHandle>& outputStrides, in computeTwoOperand()
76 const std::vector<ExprHandle>& outputStrides, in computeTwoOperandWithAlpha()
102 const std::vector<ExprHandle>& outputStrides, in computeConditionWithTwoOperand()
131 const std::vector<ExprHandle>& outputStrides, in computeThreeOperand()
161 const std::vector<ExprHandle>& outputStrides, in computeFourOperand()
191 const std::vector<ExprHandle>& outputStrides, in computeNoop()
207 const std::vector<ExprHandle>& outputStrides, in computeScalar()
H A Dmisc.cpp321 const std::vector<ExprHandle>& outputStrides, in computeChunk()
355 const std::vector<ExprHandle>& outputStrides, in computeTranspose()
382 const std::vector<ExprHandle>& outputStrides, in computeExpand()
396 const std::vector<ExprHandle>& outputStrides, in computeReshape()
463 const std::vector<ExprHandle>& outputStrides, in computeFlatten()
511 const std::vector<ExprHandle>& outputStrides) { in computeCatWoConditionals()
620 const std::vector<ExprHandle>& outputStrides, in computeCat()
682 const std::vector<ExprHandle>& outputStrides, in computeEmbedding()
H A Dreduction.cpp23 const std::vector<ExprHandle>& outputStrides, in computeSum()
108 const std::vector<ExprHandle>& outputStrides, in computeMean()
140 const std::vector<ExprHandle>& outputStrides, in computeMax()
164 const std::vector<ExprHandle>& outputStrides, in computeAdaptiveAvgPool2d()
H A Dconv2d.cpp351 const std::vector<ExprHandle>& outputStrides, in computeConv2d()
399 const std::vector<ExprHandle>& outputStrides, in computeConv1d()
433 const std::vector<ExprHandle>& outputStrides, in computePrepackedConv2dClampRun()
452 const std::vector<ExprHandle>& outputStrides, in computePrepackedLinearClampRun()
471 const std::vector<ExprHandle>& outputStrides, in computeMkldnnPrepackedConvRun()
H A Dmatmul.cpp9 const std::vector<ExprHandle>& outputStrides, in computeMatmul()
56 const std::vector<ExprHandle>& outputStrides, in computeAddMM()
H A Dnorm.cpp9 const std::vector<ExprHandle>& outputStrides, in computeBatchNorm()
H A Dsoftmax.cpp10 const std::vector<ExprHandle>& outputStrides, in computeSoftmax()
/aosp_15_r20/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
H A DTensorConvolution.h40 array<Index, NumDims> outputStrides; in IndexMapper() local
/aosp_15_r20/external/pytorch/torch/csrc/jit/codegen/onednn/
H A Dgraph_helper.cpp305 auto outputStrides = typeOfOutput->strides().concrete_sizes(); in createOperator() local
/aosp_15_r20/external/pytorch/test/cpp/tensorexpr/
H A Dtest_kernel.cpp1667 const std::vector<ExprHandle>& outputStrides, in lowerNanToNum()