Home
last modified time | relevance | path

Searched refs:inferExpandGeometry_dimvector (Results 1 – 5 of 5) sorted by relevance

/external/pytorch/aten/src/ATen/
DExpandUtils.cpp125 InferExpandGeometryResult<DimVector> inferExpandGeometry_dimvector( in inferExpandGeometry_dimvector() function
DExpandUtils.h48 TORCH_API InferExpandGeometryResult<DimVector> inferExpandGeometry_dimvector(
/external/pytorch/aten/src/ATen/native/mps/operations/
DLinear.mm47 inferExpandGeometry_dimvector(bias.sizes(), bias.strides(), output_size);
/external/pytorch/torch/csrc/jit/passes/
Dshape_analysis.cpp2091 auto sizesAndStrides = at::inferExpandGeometry_dimvector( in PropagateCompleteShapeOnNode()
/external/pytorch/aten/src/ATen/native/
DTensorShape.cpp1147 auto expandedSizesAndStrides = inferExpandGeometry_dimvector(self.sizes(), self.strides(), size); in expand()