Searched defs:dim_i (Results 1 – 8 of 8) sorted by relevance
82 const auto dim_i = static_cast<se::dnn::DimIndex>(i); in Compute() local218 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
277 int64_t dim_i = input_shape.dim_size(i); in ValidateStridedSliceOp() local
1829 int64_t dim_i = input_shape[i]; in CalculateSlicedShapeFromDenseIndices() local
1876 int dim_i = input_type.getShape()[i]; in Verify() local
5831 Value dim_i = rewriter.create<IndexCastOp>( in matchAndRewrite() local5839 Value dim_i = rewriter.create<ConstantOp>( in matchAndRewrite() local
741 const nvinfer1::Dims dim_i = inputs.at(i).GetTrtDims(); in VerifyShapesMatch() local