Searched refs:inferExpandGeometry_dimvector (Results 1 – 5 of 5) sorted by relevance
125 InferExpandGeometryResult<DimVector> inferExpandGeometry_dimvector( in inferExpandGeometry_dimvector() function
48 TORCH_API InferExpandGeometryResult<DimVector> inferExpandGeometry_dimvector(
47 inferExpandGeometry_dimvector(bias.sizes(), bias.strides(), output_size);
2091 auto sizesAndStrides = at::inferExpandGeometry_dimvector( in PropagateCompleteShapeOnNode()
1147 auto expandedSizesAndStrides = inferExpandGeometry_dimvector(self.sizes(), self.strides(), size); in expand()