Searched refs:result_rank (Results 1 – 4 of 4) sorted by relevance
126 int64_t result_rank = 1 + data0_shape.dims() - indices0_shape.dims(); in Compile() local128 std::vector<int64_t> result_shape(result_rank); in Compile()182 std::vector<int64_t> slice_start(result_rank); in Compile()183 std::vector<int64_t> slice_limit(result_rank); in Compile()184 std::vector<int64_t> stride(result_rank, 1); in Compile()
582 int64_t result_rank = -1; // -1 means unknown result rank. in matchAndRewrite() local588 result_rank = element_type.getRank() + 1; in matchAndRewrite()604 Type shape_type = RankedTensorType::get({result_rank}, shape_dtype); in matchAndRewrite()
3510 int64_t result_rank = gather_dim_numbers.offset_dims_size() + in InferGatherShape() local3515 output_dim_bounds.reserve(result_rank); in InferGatherShape()3518 output_dim_is_dynamic.reserve(result_rank); in InferGatherShape()3519 for (int64_t i = 0; i < result_rank; i++) { in InferGatherShape()
3218 auto result_rank = result_ty.getRank(); in matchAndRewrite() local3239 multiply_constant_ints.resize(result_rank, 1); in matchAndRewrite()3240 for (int i = result_rank - 1; i > 0; i--) { in matchAndRewrite()3250 RankedTensorType::get({result_rank}, indices_ety); in matchAndRewrite()