Searched defs:dim_i (Results 1 – 8 of 8) sorted by relevance
83 const auto dim_i = static_cast<se::dnn::DimIndex>(i); in Compute() local213 const auto dim_i = static_cast<se::dnn::DimIndex>(i); in Compute() local
109 inline int Dim(int dim_i) const { in Dim()
64 int dim_h, dim_w, dim_i, dim_o; member
281 int64_t dim_i = input_shape.dim_size(i); in ValidateStridedSliceOp() local
1914 int64_t dim_i = input_shape[i]; in CalculateSlicedShapeFromDenseIndices() local
5577 Value dim_i = rewriter.create<arith::IndexCastOp>( in matchAndRewrite() local5585 Value dim_i = rewriter.create<arith::ConstantOp>( in matchAndRewrite() local
2054 int dim_i = input_type.getShape()[i]; in verify() local
640 const nvinfer1::Dims dim_i = inputs.at(i).GetTrtDims(); in VerifyShapesMatch() local