Searched refs:num_input_dims (Results 1 – 3 of 3) sorted by relevance
452 const int32_t num_input_dims = input_shape.DimensionsCount(); in EvalHybrid() local456 const int input_size = input_shape.Dims(num_input_dims - 2); in EvalHybrid()457 const int batch_size = input_shape.Dims(num_input_dims - 1); in EvalHybrid()
793 int num_input_dims = ranked_input_type.getRank(); in matchAndRewrite() local794 SmallVector<int32_t, 4> padding_begin(num_input_dims, 0); in matchAndRewrite()797 SmallVector<int32_t, 4> padding_strides(num_input_dims, 1); in matchAndRewrite()817 RankedTensorType::get({num_input_dims}, rewriter.getIntegerType(32)); in matchAndRewrite()
1785 const int num_input_dims = context->src_format.length(); in UpdateSqueezeDims() local1786 const int min_squeeze_dim = -num_input_dims; in UpdateSqueezeDims()1792 if (dim < min_squeeze_dim || dim >= num_input_dims) { in UpdateSqueezeDims()1796 num_input_dims, ")"); in UpdateSqueezeDims()1799 dim += num_input_dims; in UpdateSqueezeDims()