Searched refs:result_rank (Results 1 – 2 of 2) sorted by relevance
116 int64 result_rank = 1 + data0_shape.dims() - indices0_shape.dims(); in Compile() local118 std::vector<int64> result_shape(result_rank); in Compile()172 std::vector<int64> slice_start(result_rank); in Compile()173 std::vector<int64> slice_limit(result_rank); in Compile()174 std::vector<int64> stride(result_rank, 1); in Compile()
3038 int64 result_rank = gather_dim_numbers.offset_dims_size() + in InferGatherShape() local3043 output_dim_bounds.reserve(result_rank); in InferGatherShape()3044 for (int64 i = 0; i < result_rank; i++) { in InferGatherShape()